Update app.py

app.py

@@ -14,7 +14,6 @@ def __():
 def __():
     import pandas as pd 
     df = pd.read_csv("datasets/test_set.csv")
-    df.head()
     return df, pd
 
 
@@ -30,7 +29,7 @@ def __():
     import torch 
     import os 
 
-    def get_models(root="/our_visualization/models"):
+    def get_models(root="models"):
         paths = os.listdir(root)
         model_paths = []
         for path in paths: