Spaces:
Runtime error
Runtime error
Commit
·
c600fb7
1
Parent(s):
cc4331b
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
4 |
tokenizer = AutoTokenizer.from_pretrained("TheCraftySlayer/Llama-2-70b-chat-hf")
|
5 |
model = AutoModelForCausalLM.from_pretrained("TheCraftySlayer/Llama-2-70b-chat-hf")
|
6 |
input_text =" hello how are you?"
|
7 |
-
inputs= tokenizer.encode(input_text, return_tensors='pt')
|
8 |
outputs= model.generate(inputs,max_length=50,num_return_sequences=5,temperature=0.7)
|
9 |
print("Generated Text")
|
10 |
for i, output in enumerate(outputs):
|
|
|
4 |
tokenizer = AutoTokenizer.from_pretrained("TheCraftySlayer/Llama-2-70b-chat-hf")
|
5 |
model = AutoModelForCausalLM.from_pretrained("TheCraftySlayer/Llama-2-70b-chat-hf")
|
6 |
input_text =" hello how are you?"
|
7 |
+
inputs= tokenizer.encode(input_text, return_tensors='pt')
|
8 |
outputs= model.generate(inputs,max_length=50,num_return_sequences=5,temperature=0.7)
|
9 |
print("Generated Text")
|
10 |
for i, output in enumerate(outputs):
|