Update app.py

app.py

@@ -17,6 +17,16 @@ import torch.nn.functional as F
 #from torch.nn import CrossEntropyLoss
 #from sklearn.decomposition import PCA
 import matplotlib.pyplot as plt
+import json
+
+# Load configuration file 
+with open('config.json', 'r') as config_file:
+    config = json.load(config_file) 
+arg1 = config.get('arg1', 'default_value1') 
+arg2 = config.get('arg2', 'default_value2') 
+
+print(f"Argument 1: {arg1}") 
+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.