SAUL19 commited on
Commit
4115131
·
1 Parent(s): 046df59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,4 +1,6 @@
1
  import gradio as gr
 
 
2
  import torch
3
  from diffusers import StableDiffusionPipeline
4
  import boto3
@@ -52,5 +54,5 @@ def text_to_image(summary, image_name):
52
 
53
 
54
 
55
- iface = gr.Interface(fn=text_to_image, inputs=[gr.inputs.Text(label="Summary"), gr.inputs.Text(label="Image Name")], outputs="text")
56
  iface.launch()
 
1
  import gradio as gr
2
+ from gradio.inputs import Textbox
3
+
4
  import torch
5
  from diffusers import StableDiffusionPipeline
6
  import boto3
 
54
 
55
 
56
 
57
+ iface = gr.Interface(fn=text_to_image, inputs=[Textbox(label="Summary"), Textbox(label="Image Name")],, outputs="text")
58
  iface.launch()