SmartHeal commited on
Commit
ccfac38
·
verified ·
1 Parent(s): 2b90bf7

Update src/ai_processor.py

Browse files
Files changed (1) hide show
  1. 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,