fokan commited on
Commit
bf46f16
·
verified ·
1 Parent(s): bbc388a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,14 +6,14 @@ import torch
6
  # ---------------------------
7
  # Load encoder (MedSigLIP)
8
  # ---------------------------
9
- ENCODER_ID = "fokan/medsiglip-448-fp16-pruned20"
10
  encoder_processor = AutoProcessor.from_pretrained(ENCODER_ID)
11
  encoder_model = AutoModel.from_pretrained(ENCODER_ID).eval()
12
 
13
  # ---------------------------
14
  # Load decoder (MedGemma)
15
  # ---------------------------
16
- DECODER_ID = "fokan/medgemma-4b-it-fp16-pruned20"
17
  decoder = pipeline("text-generation", model=DECODER_ID, torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32)
18
 
19
  # ---------------------------
 
6
  # ---------------------------
7
  # Load encoder (MedSigLIP)
8
  # ---------------------------
9
+ ENCODER_ID = "fokan/medsiglip-448-int8"
10
  encoder_processor = AutoProcessor.from_pretrained(ENCODER_ID)
11
  encoder_model = AutoModel.from_pretrained(ENCODER_ID).eval()
12
 
13
  # ---------------------------
14
  # Load decoder (MedGemma)
15
  # ---------------------------
16
+ DECODER_ID = "fokan/medgemma-4b-it-int8"
17
  decoder = pipeline("text-generation", model=DECODER_ID, torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32)
18
 
19
  # ---------------------------