Update app.py

app.py

@@ -7,6 +7,7 @@ import zipfile
 import os
 import torch
 
+
 # Percorsi ZIP per manuali e problemi
 zip_path_m = "faiss_manual_index.zip"
 faiss_manual_index = "faiss_manual_index"
@@ -31,6 +32,15 @@ problems_vectorstore = FAISS.load_local(faiss_problems_index, embedding_model, a
 
 # Caricamento del modello GPT-J da Hugging Face
 model_name = "EleutherAI/gpt-j-6B"
+
+# Forza l'uso della CPU
+model = AutoModelForCausalLM.from_pretrained(
+    model_name, 
+    torch_dtype=torch.float32,  # float32 per la CPU
+    device_map={"": "cpu"}  # Specifica CPU come dispositivo
+)
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")