weyavon578 commited on
Commit
cef42b5
·
verified ·
1 Parent(s): 1c33724

Update application.py

Browse files
Files changed (1) hide show
  1. application.py +0 -15
application.py CHANGED
@@ -1,17 +1,2 @@
1
  pip install transformers torch
2
- from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
- # Nome do modelo que você deseja carregar
5
- model_name = "EleutherAI/gpt-j-6B" # Exemplo de modelo suportado
6
-
7
- try:
8
- model = AutoModelForCausalLM.from_pretrained(model_name)
9
- tokenizer = AutoTokenizer.from_pretrained(model_name)
10
-
11
- input_text = "O que você pensa sobre aprendizado de máquina?"
12
- inputs = tokenizer(input_text, return_tensors="pt")
13
- outputs = model(**inputs)
14
-
15
- print(outputs)
16
- except Exception as e:
17
- print(f"Erro ao carregar o modelo: {e}")
 
1
  pip install transformers torch
 
2