Create application.py

application.py

@@ -1,2 +1,16 @@
-pip install transformers torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
+# Nome do modelo que você deseja carregar
+model_name = "EleutherAI/gpt-j-6B"  # Exemplo de modelo suportado
+
+try:
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+    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}")