Update app.py

app.py

@@ -8,8 +8,8 @@ from matplotlib import pyplot as plt
 
 
 def gradio_demo(model_name, sequence_input, nucleus_image, protein_image):
-    model = f"models/{model_name}.ckpt"
-    config = f"models/{model_name}.yaml"
+    model = f"CELL-E_2-Image_Prediction/models/{model_name}.ckpt"
+    config = f"CELL-E_2-Image_Prediction/models/{model_name}.yaml"
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")