Commit
•
f65befc
1
Parent(s):
c5e75f3
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ def greet(txt):
|
|
8 |
"text-generation", model="checkpoint-1500", device=device
|
9 |
)
|
10 |
|
11 |
-
return
|
12 |
|
13 |
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
|
14 |
|
|
|
8 |
"text-generation", model="checkpoint-1500", device=device
|
9 |
)
|
10 |
|
11 |
+
return pipe(txt, num_return_sequences=1)[0]["generated_text"]
|
12 |
|
13 |
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
|
14 |
|