Update app.py

app.py

@@ -1,10 +1,18 @@
 import gradio as gr
-import clinicadl
+from huggingface_hub import hf_hub_download
+import torch
 
+# Download model from Hub
+model_path = hf_hub_download(repo_id="ARAMIS-LAB/CNN-AD-CN", filename="model.pth.tar")
 
-def greet(image):
-    
-    return "Hello " + name + "!!"
+# Load ClinicaDL model
+model = torch.load(model_path, map_location="cpu")
+model.eval()
 
-demo = gr.Interface(fn=greet, inputs="image", outputs="text")
-demo.launch()
+def predict(input_image):
+    with torch.no_grad():
+        output = model(input_image.unsqueeze(0))
+    return output.numpy().tolist()
+
+demo = gr.Interface(fn=predict, inputs="image", outputs="label")
+demo.launch()