Update app.py

app.py

@@ -6,8 +6,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 MODEL_NAME = "deepseek-ai/deepseek-R1"  # Verificar nome exato no Hugging Face Hub
 
 # Carregar tokenizer e modelo
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, trust_remote_code=True,  revision="main")
 
 # Configurações de comprimento
 TARGET_LENGTH = 256