Update app.py

app.py

@@ -15,7 +15,9 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load the default model initially
 current_model_name = "DialoGPT-med-FT"
-model = AutoModelForCausalLM.from_pretrained(model_names[current_model_name]).to(device)
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
+model.load_state_dict(torch.load(model_names[current_model_name], map_location=device))
+model.to(device)
 
 def load_model(model_name):
     global model, current_model_name