prithivMLmods commited on
Commit
86b0f3f
·
verified ·
1 Parent(s): 1203426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -113,7 +113,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
113
  buffer = ""
114
  for new_text in streamer:
115
  buffer += new_text
116
- buffer = buffer.replace("<|im_end|>", "")
117
  time.sleep(0.01)
118
  yield buffer
119
 
@@ -233,7 +233,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
233
  model_choice = gr.Radio(
234
  choices=["DREX-062225-exp", "VIREX-062225-exp"],
235
  label="Select Model",
236
- value="VIREX-062225-exp"
237
  )
238
 
239
  gr.Markdown("**Model Info 💻** | [Report Bug](https://huggingface.co/spaces/prithivMLmods/Doc-VLMs/discussions)")
 
113
  buffer = ""
114
  for new_text in streamer:
115
  buffer += new_text
116
+ #buffer = buffer.replace("<|im_end|>", "")
117
  time.sleep(0.01)
118
  yield buffer
119
 
 
233
  model_choice = gr.Radio(
234
  choices=["DREX-062225-exp", "VIREX-062225-exp"],
235
  label="Select Model",
236
+ value="DREX-062225-exp"
237
  )
238
 
239
  gr.Markdown("**Model Info 💻** | [Report Bug](https://huggingface.co/spaces/prithivMLmods/Doc-VLMs/discussions)")