LLDDWW Claude commited on
Commit
7d833fa
ยท
1 Parent(s): d5aff0d

chore: upgrade to MedGemma-4B-IT for medical analysis

Browse files

- Switch from Gemma-2-2B to MedGemma-4B-IT
- MedGemma is specifically trained for medical information
- Better medication analysis and explanations

๐Ÿค– Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ from qwen_vl_utils import process_vision_info
14
  OCR_MODEL_ID = "Qwen/Qwen2.5-VL-7B-Instruct"
15
 
16
  # ์•ฝ ์ •๋ณด ๋ถ„์„ ๋ชจ๋ธ ID (์˜๋ฃŒ ์ „๋ฌธ)
17
- MED_MODEL_ID = "google/gemma-2-2b-it"
18
 
19
 
20
  def _extract_assistant_content(decoded: str) -> str:
@@ -370,7 +370,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
370
 
371
  **๐Ÿค– ๊ธฐ์ˆ  ์Šคํƒ**
372
  - Qwen2.5-VL-7B-Instruct (OCR ํ…์ŠคํŠธ ์ถ”์ถœ)
373
- - Google Gemma-2-2B-IT (์˜๋ฃŒ ์ •๋ณด ๋ถ„์„ ๋ฐ ์„ค๋ช…)
374
  """)
375
 
376
  if __name__ == "__main__":
 
14
  OCR_MODEL_ID = "Qwen/Qwen2.5-VL-7B-Instruct"
15
 
16
  # ์•ฝ ์ •๋ณด ๋ถ„์„ ๋ชจ๋ธ ID (์˜๋ฃŒ ์ „๋ฌธ)
17
+ MED_MODEL_ID = "google/medgemma-4b-it"
18
 
19
 
20
  def _extract_assistant_content(decoded: str) -> str:
 
370
 
371
  **๐Ÿค– ๊ธฐ์ˆ  ์Šคํƒ**
372
  - Qwen2.5-VL-7B-Instruct (OCR ํ…์ŠคํŠธ ์ถ”์ถœ)
373
+ - Google MedGemma-4B-IT (์˜๋ฃŒ ์ „๋ฌธ ๋ชจ๋ธ - ์•ฝ ์ •๋ณด ๋ถ„์„ ๋ฐ ์„ค๋ช…)
374
  """)
375
 
376
  if __name__ == "__main__":