Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,6 +14,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
|
| 14 |
|
| 15 |
# demo = gr.Interface(fn=greet, inputs=["text"], outputs=["text"],)
|
| 16 |
|
|
|
|
| 17 |
def greet(input_text):
|
| 18 |
input_token = tokenizer.encode(input_text, return_tensors="pt").to("cpu")
|
| 19 |
|
|
|
|
| 14 |
|
| 15 |
# demo = gr.Interface(fn=greet, inputs=["text"], outputs=["text"],)
|
| 16 |
|
| 17 |
+
@spaces.GPU
|
| 18 |
def greet(input_text):
|
| 19 |
input_token = tokenizer.encode(input_text, return_tensors="pt").to("cpu")
|
| 20 |
|