Unityraptor commited on
Commit
ea0b887
1 Parent(s): ab9ad6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,8 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipe = pipeline("image-to-text",
5
- model="Salesforce/blip-image-captioning-base")
6
 
7
 
8
  def launch(input):
@@ -10,7 +9,7 @@ def launch(input):
10
  return out[0]['generated_text']
11
 
12
  iface = gr.Interface(launch,
13
- inputs=gr.Image(type='pil'),
14
  outputs="text")
15
 
16
  iface.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline("text2text-generation", model="humarin/chatgpt_paraphraser_on_T5_base")
 
5
 
6
 
7
  def launch(input):
 
9
  return out[0]['generated_text']
10
 
11
  iface = gr.Interface(launch,
12
+ inputs="textbox",
13
  outputs="text")
14
 
15
  iface.launch()