Omnibus commited on
Commit
aff77e1
·
verified ·
1 Parent(s): c2a4c31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,7 +64,7 @@ def generate(prompt, history):
64
  pass
65
  '''
66
  output=output.replace("\n"," ")
67
- if "PROMPT:" & "MEME_TEXT:" in output:
68
  output={'PROMPT':output.split("PROMPT:",1)[1].split("MEME_TEXT",1)[0],'MEME_TEXT':output.split("MEME_TEXT",1)[1]}
69
  output=str(output)
70
  return [(prompt,output)]
@@ -146,7 +146,7 @@ with gr.Blocks() as app:
146
  im_out=gr.Image(label="Image")
147
 
148
  sub_b = submit_b.click(generate, [msg,chatbot],[chatbot])
149
- sub_im = submit_b.click(run, [msg,chatbot,model_drop],[chatbot,im_out])
150
  sub_e = msg.submit(run, [msg, chatbot,model_drop], [chatbot,im_out])
151
  stop_b.click(None,None,None, cancels=[sub_b,sub_e])
152
  app.launch()
 
64
  pass
65
  '''
66
  output=output.replace("\n"," ")
67
+ if "PROMPT:" and "MEME_TEXT:" in output:
68
  output={'PROMPT':output.split("PROMPT:",1)[1].split("MEME_TEXT",1)[0],'MEME_TEXT':output.split("MEME_TEXT",1)[1]}
69
  output=str(output)
70
  return [(prompt,output)]
 
146
  im_out=gr.Image(label="Image")
147
 
148
  sub_b = submit_b.click(generate, [msg,chatbot],[chatbot])
149
+ sub_im = submit_im.click(run, [msg,chatbot,model_drop],[chatbot,im_out])
150
  sub_e = msg.submit(run, [msg, chatbot,model_drop], [chatbot,im_out])
151
  stop_b.click(None,None,None, cancels=[sub_b,sub_e])
152
  app.launch()