Update app.py

app.py

@@ -6,7 +6,7 @@ from huggingface_hub import hf_hub_download
 import gradio as gr
 
 # Download model from Hugging Face Hub
-model_path = hf_hub_download(repo_id="Ayamohamed/DiaClassification", filename="dia_class_full.pth")
+model_path = hf_hub_download(repo_id="Ayamohamed/DiaClassification", filename="model.pth")
 
 # Load model
 model = torch.load(model_path,weights_only=False)