Nitinnn commited on
Commit
e3fbad4
1 Parent(s): 0e1d9ba

Update app.py

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