Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import requests
+from diffusers import DiffusionPipeline
 
 
 
@@ -17,10 +18,11 @@ def translate(hin_snippet):
     
 
     translated = tokenizer.decode(outputs[0]).replace('<pad>',"").strip().lower()
-    r = requests.post(url='https://hf.space/embed/multimodalart/latentdiffusion/+/api/predict/',
-        json={"data":[translated,50,32,32,4,15.0]})
-    print(r.json())
-    return r.json()
+
+    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")
 
 
 tokenizer = AutoTokenizer.from_pretrained("salesken/translation-hi-en")
@@ -32,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,gr.outputs.Carousel(label="Individual images",components=["image"]),gr.outputs.Textbox(label="Error")], css=css)
+iface = gr.Interface(fn=translate, inputs="text",outputs="image")
 iface.launch()