Update app.py

app.py

@@ -76,6 +76,19 @@ def generate(prompt, history):
     yield [(prompt,output)],output1
     
 def run(inp,model_drop):
+    prompt=inp['PROMPT']
+    model=loaded_model[int(model_drop)]
+    out_img=model(resp)
+    print(out_img)
+    url=f'https://omnibus-meme-diffusion.hf.space/file={out_img}'
+    print(url)
+    uid = uuid.uuid4()
+    r = requests.get(url, stream=True)
+    if r.status_code == 200:
+        out = Image.open(io.BytesIO(r.content))
+    return out
+
+def run_OG(inp,model_drop):
     prompt=inp['PROMPT']
     #if len(inp)>max_prompt:
     #    inp = condense(inp)