Spaces:
Running
Running
Update src/ai_processor.py
Browse files- src/ai_processor.py +1 -0
src/ai_processor.py
CHANGED
|
@@ -161,6 +161,7 @@ def _vlm_infer_gpu(messages, model_id: str, max_new_tokens: int, token: Optional
|
|
| 161 |
txt = out[0].get("generated_text", "")
|
| 162 |
return (txt or "").strip() or "⚠️ Empty response"
|
| 163 |
|
|
|
|
| 164 |
def generate_medgemma_report( # kept name so callers don't change
|
| 165 |
patient_info: str,
|
| 166 |
visual_results: Dict,
|
|
|
|
| 161 |
txt = out[0].get("generated_text", "")
|
| 162 |
return (txt or "").strip() or "⚠️ Empty response"
|
| 163 |
|
| 164 |
+
@_SPACES_GPU(enable_queue=True)
|
| 165 |
def generate_medgemma_report( # kept name so callers don't change
|
| 166 |
patient_info: str,
|
| 167 |
visual_results: Dict,
|