Spaces:
Sleeping
Sleeping
Update src/ai_processor.py
Browse files- 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=
|
| 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).
|