Update app.py

app.py

@@ -18,11 +18,14 @@ models = {
 
 loaded_models = {}
 
+# Ensure we're using the CPU if GPU isn't available or necessary
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 def load_model(model_name):
     if model_name not in loaded_models:
-        loaded_models[model_name] = pipeline("question-answering", model=models[model_name])
+        loaded_models[model_name] = pipeline("question-answering", model=models[model_name], device=0 if torch.cuda.is_available() else -1)
     return loaded_models[model_name]
-
+    
 def generate_score_chart(score):
     plt.figure(figsize=(6, 4))
     plt.bar(["Confidence Score"], [score], color='skyblue')