SmartHeal commited on
Commit
d72cd8b
·
verified ·
1 Parent(s): a64b146

Update src/ai_processor.py

Browse files
Files changed (1) hide show
  1. src/ai_processor.py +1 -1
src/ai_processor.py CHANGED
@@ -183,7 +183,7 @@ Disclaimer
183
  # ---------- MedGemma-only text generator ----------
184
  @_SPACES_GPU(enable_queue=True)
185
  def vlm_generate(prompt, image_pil, model_id="unsloth/medgemma-4b-it-bnb-4bit",
186
- max_new_tokens=256, token=None):
187
  """
188
  Simple helper: messages-style image+text → text using a 4-bit MedGemma pipeline.
189
  - No explicit `device` argument (pipeline will auto-detect).
 
183
  # ---------- MedGemma-only text generator ----------
184
  @_SPACES_GPU(enable_queue=True)
185
  def vlm_generate(prompt, image_pil, model_id="unsloth/medgemma-4b-it-bnb-4bit",
186
+ max_new_tokens=1024, token=None):
187
  """
188
  Simple helper: messages-style image+text → text using a 4-bit MedGemma pipeline.
189
  - No explicit `device` argument (pipeline will auto-detect).