Update app.py

app.py

@@ -504,11 +504,11 @@ class MyNetTest(nn.Module):
 
 
 model = MyNet(emb_dim=512, feat_dim=512)
-state = torch.load('./best_state_download_dict.pth')
+state = torch.load(os.path.join(os.getcwd(),'best_state_download_dict.pth'))
 model.load_state_dict(state)
 model.eval()
 try:
-    os.mkdir('./save_df/')
+    os.mkdir(os.path.join(os.getcwd(),'save_df'))
 except:
     pass
 
@@ -522,7 +522,7 @@ def get_rt_from_mol(mol):
 def pred_file_btyes(file_bytes,progress=gr.Progress()):
     progress(0,desc='Starting')
     file_name = os.path.join(
-        './save_df/',
+        os.path.join(os.getcwd(),'save_df'),
         (hashlib.md5(str(file_bytes).encode('utf-8')).hexdigest()+'.csv')
         )
     if os.path.exists(file_name):