Spaces:

tamatwi
/

shortrinnashortgpu_rinna

App Files Files Community

tamatwi commited on Jun 25

Commit

dd09039

•

1 Parent(s): ca66480

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 import spaces
-# @spaces.GPU
 # generator = pipeline('text-generation', model='distilgpt2', device=-1)
 def generate_text(prompt, max_length):

 import gradio as gr
 from transformers import pipeline
 import spaces
+@spaces.GPU
 # generator = pipeline('text-generation', model='distilgpt2', device=-1)
 def generate_text(prompt, max_length):