Update app.py

app.py

@@ -3,18 +3,21 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Configurações do modelo DeepSeek-R1
-MODEL_NAME = "deepseek-ai/deepseek-R1"  # Verificar nome exato no Hugging Face Hub
+MODEL_NAME = "deepseek-ai/deepseek-R1"
 
-# Carregar tokenizer e modelo
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME,
-                                             trust_remote_code=True, 
-                                             revision="main",
-                                             device_map="auto",
-                                             torch_dtype=torch.float32,
-                                             load_in_8bit=False,
-                                             low_cpu_mem_usage=True
-                                            )
+# Carregar tokenizer e modelo; Configurar para CPU
+tokenizer = AutoTokenizer.from_pretrained(
+    MODEL_NAME,
+    trust_remote_code=True
+)
+
+model = AutoModelForCausalLM.from_pretrained(
+    MODEL_NAME,
+    trust_remote_code=True,
+    torch_dtype=torch.float32,
+    device_map="auto",
+    low_cpu_mem_usage=True
+)
 
 # Configurações de comprimento
 TARGET_LENGTH = 256