austinroy commited on
Commit
48aca08
·
1 Parent(s): 5a49f1b

Initial commit

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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