Update app.py

app.py

@@ -9,7 +9,7 @@ import streamlit as st
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 # Load the trained model from the Hugging Face Hub
-model = AutoModel.from_pretrained('dhhd255/parkinsons_pred0.1')
+model = AutoModel.from_pretrained('dhhd255/parkinsons_20epochs')
 
 # Move the model to the device
 model = model.to(device)