Update load_model.py

load_model.py

@@ -1,16 +1,15 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Nome do modelo correto
-model_name = "gpt2"  # ou "EleutherAI/gpt-j-6B" ou outro modelo suportado
+model_name = "EleutherAI/gpt-j-6B"  # Exemplo de modelo suportado
 
-# Carregar o modelo
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+try:
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-# Testar o modelo com uma entrada simples
-input_text = "O que você acha sobre inteligência artificial?"
-inputs = tokenizer(input_text, return_tensors="pt")
-outputs = model(**inputs)
-
-print(outputs)
+    input_text = "O que você pensa sobre aprendizado de máquina?"
+    inputs = tokenizer(input_text, return_tensors="pt")
+    outputs = model(**inputs)
 
+    print(outputs)
+except Exception as e:
+    print(f"Erro ao carregar o modelo: {e}")