Spaces:
Runtime error
Runtime error
Initial commit
Browse files
app.py
CHANGED
@@ -22,6 +22,7 @@ from google.genai import types
|
|
22 |
import os
|
23 |
|
24 |
gemini_api_key = os.getenv('GEMINI_API_KEY')
|
|
|
25 |
client = genai.Client(api_key=gemini_api_key)
|
26 |
|
27 |
"""### 🧠 Loading MedGemma for Radiology Insights
|
@@ -33,7 +34,7 @@ from transformers import pipeline, BitsAndBytesConfig
|
|
33 |
import torch
|
34 |
|
35 |
model_kwargs = dict(torch_dtype=torch.bfloat16, device_map="cuda:0", quantization_config=BitsAndBytesConfig(load_in_4bit=True))
|
36 |
-
pipe = pipeline("image-text-to-text", model="google/medgemma-4b-it", model_kwargs=model_kwargs)
|
37 |
pipe.model.generation_config.do_sample = False
|
38 |
|
39 |
"""### 🩻 Radiology Image Interpretation Logic
|
|
|
22 |
import os
|
23 |
|
24 |
gemini_api_key = os.getenv('GEMINI_API_KEY')
|
25 |
+
hf_token = os.environ.get("HF_TOKEN")
|
26 |
client = genai.Client(api_key=gemini_api_key)
|
27 |
|
28 |
"""### 🧠 Loading MedGemma for Radiology Insights
|
|
|
34 |
import torch
|
35 |
|
36 |
model_kwargs = dict(torch_dtype=torch.bfloat16, device_map="cuda:0", quantization_config=BitsAndBytesConfig(load_in_4bit=True))
|
37 |
+
pipe = pipeline("image-text-to-text", model="google/medgemma-4b-it", model_kwargs=model_kwargs, token=hf_token)
|
38 |
pipe.model.generation_config.do_sample = False
|
39 |
|
40 |
"""### 🩻 Radiology Image Interpretation Logic
|