Create load_model.py

load_model.py

@@ -0,0 +1,16 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# Nome do modelo correto
+model_name = "gpt2"  # ou "EleutherAI/gpt-j-6B" ou outro modelo suportado
+
+# Carregar o modelo
+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)
+