change argument script

app.py

@@ -30,19 +30,19 @@ print(f"Argument 2: {arg2}")
 
 if len(sys.argv) > 1:
     # sys.argv[0] is the script name, sys.argv[1] is the first argument, etc.
-    runModel = sys.argv[1]
+    runModel = arg1
     print(f"Passed value: {runModel}")
-    print (sys.argv[2])
+    print (arg2)
     
 else:
     print("No argument was passed.")
 
 
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
-modelNameToUse = sys.argv[2]
+modelNameToUse = arg2
 
 if (runModel=='1'):
-    dataFileName = sys.argv[2] + '.csv'
+    dataFileName = arg2 + '.csv'
     print (dataFileName)
     # Load the data from the CSV file
     df = pd.read_csv(dataFileName)