turtlegraphics commited on
Commit
1609e16
·
verified ·
1 Parent(s): 804d2ec

Now using predictive text.

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -2,10 +2,14 @@
2
  # gradio demo
3
  #
4
  import gradio as gr
 
5
 
6
- def greet(name):
7
- return "Hello " + name
8
 
9
- demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 
10
 
11
- demo.launch()
 
 
 
2
  # gradio demo
3
  #
4
  import gradio as gr
5
+ from transformers import pipeline
6
 
7
+ model = pipeline("text-generation")
 
8
 
9
+ def predict(prompt):
10
+ completion = model(prompt)[0]["generated_text"]
11
+ return completion
12
 
13
+ demo = gr.Interface(fn=predict, inputs="text", outputs="text")
14
+
15
+ demo.launch()