Update app.py

app.py

@@ -15,7 +15,7 @@ def resize(value,img):
     img = img.resize((value,value))
     return img
 
-def predict(source_img, prompt, strength):
+def img2img(source_img, prompt, strength):
     imageio.imwrite("data.png", source_img['image'])
     imageio.imwrite("data_mask.png", source_img["mask"])
     src = resize(768, "data.png")
@@ -24,10 +24,14 @@ def predict(source_img, prompt, strength):
     mask.save("mask.png")
     image = pipe(prompt=prompt, image=src, mask_image=mask, num_inference_steps=6, strength=strength, guidance_scale=0.0).images[0]
     return image
-
+    
+def txt2img(prompt):
+    image=pipe(prompt=prompt, num_inference_steps=3, guidance_scale=0.0).images[0]
+    return image
+    
 title="SDXL Turbo Inpainting CPU"
 description="Inpainting with SDXL Turbo <br><br> <b>Please use square .png image as input, 512x512, 768x768, or 1024x1024</b>"
-gr.Interface(fn=predict, inputs=[gr.Image(source=("upload"), tool='sketch', label='Source Image'), 
+gr.Interface(fn=[img2img, txt2img], inputs=[gr.Image(source=("upload"), tool='sketch', label='Source Image'), 
                                 gr.Textbox(label='What you want the AI to Generate, 77 Token limit'),
                                 gr.Slider(minimum=.5, maximum=1, value=.75, step=.025, label='Strength')], 
              outputs='image',