Spaces:

tamatwi
/

shortrinnashortgpu_rinna

App Files Files Community

tamatwi commited on Jun 25

Commit

ca66480

•

1 Parent(s): f81ad1c

Update app.py

Files changed (1) hide show

app.py +2 -2

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
 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):
     result = generator(prompt, max_length=max_length, num_return_sequences=1)

 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):
     result = generator(prompt, max_length=max_length, num_return_sequences=1)