Update app.py

app.py

@@ -22,7 +22,7 @@ def translate(hin_snippet):
     model_id = "CompVis/ldm-text2im-large-256"
     ldm = DiffusionPipeline.from_pretrained(model_id)
     images = ldm([translated], num_inference_steps=50, eta=0.3, guidance_scale=6)["sample"]
-    return images[0].save(f"out.png")
+    return images
 
 
 tokenizer = AutoTokenizer.from_pretrained("salesken/translation-hi-en")
@@ -34,5 +34,5 @@ model = AutoModelForSeq2SeqLM.from_pretrained("salesken/translation-hi-en")
 # due to covid, we have reduced our debt interest
 
 
-iface = gr.Interface(fn=translate, inputs="text",outputs="image")
+iface = gr.Interface(fn=translate, inputs="text",outputs=gr.Gallery())
 iface.launch()