Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from huggingface_hub import hf_hub_download
 from prediction import run_sequence_prediction
 import torch
 import torchvision.transforms as T
@@ -9,11 +8,8 @@ from matplotlib import pyplot as plt
 
 
 def gradio_demo(model_name, sequence_input, image):
-    model = hf_hub_download(repo_id=f"HuangLab/{model_name}", filename="model.ckpt")
-    config = hf_hub_download(repo_id=f"HuangLab/{model_name}", filename="config.yaml")
-    hf_hub_download(repo_id=f"HuangLab/{model_name}", filename="nucleus_vqgan.yaml")
-    hf_hub_download(repo_id=f"HuangLab/{model_name}", filename="threshold_vqgan.yaml")
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    model = f"CELL-E_2-Image_Prediction/models/{model_name}.ckpt"
+    config = f"CELL-E_2-Image_Prediction/models/{model_name}.yaml"
 
     if "Finetuned" in model_name:
         dataset = "OpenCell"