Update app.py

app.py

@@ -426,8 +426,10 @@ def get_prompts(prompt_text):
     return gen   
 
 def gen_im(prompt,model_choice=5,count=1):
-    out = dif(prompt,5,count)
-    return out
+    out = dif.SendIt()
+    ou1 = out.send_it(prompt,5,count)
+
+    return out1
     
     
 with gr.Blocks() as build_app: