visheratin commited on
Commit
93c40a1
·
verified ·
1 Parent(s): e0c666b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -67,8 +67,9 @@ def answer_question(image, question, max_crops, num_tokens):
67
  buffer = ""
68
  for new_text in streamer:
69
  buffer += new_text
 
70
  if len(buffer) > 1:
71
- yield buffer.replace(prompt, "")
72
 
73
 
74
  with gr.Blocks() as demo:
@@ -95,4 +96,4 @@ with gr.Blocks() as demo:
95
  submit.click(answer_question, [img, prompt, max_crops, num_tokens], output)
96
  prompt.submit(answer_question, [img, prompt, max_crops, num_tokens], output)
97
 
98
- demo.queue().launch(debug=True, share=True)
 
67
  buffer = ""
68
  for new_text in streamer:
69
  buffer += new_text
70
+ print(buffer)
71
  if len(buffer) > 1:
72
+ yield buffer
73
 
74
 
75
  with gr.Blocks() as demo:
 
96
  submit.click(answer_question, [img, prompt, max_crops, num_tokens], output)
97
  prompt.submit(answer_question, [img, prompt, max_crops, num_tokens], output)
98
 
99
+ demo.queue().launch(debug=True)