AfriAISolutions commited on
Commit
959b6cf
·
verified ·
1 Parent(s): 3c12b02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ login(token=hf_token)
18
  translator = transformers.pipeline("translation", model="facebook/nllb-200-distilled-600M")
19
  text_gen_pipeline = transformers.pipeline(
20
  "text-generation",
21
- model = "ruslanmv/Medical-Llama3-v2",
22
- # model="ContactDoctor/Bio-Medical-Llama-3-2-1B-CoT-012025",
23
  torch_dtype=torch.bfloat16,
24
  device_map="auto"
25
  )
 
18
  translator = transformers.pipeline("translation", model="facebook/nllb-200-distilled-600M")
19
  text_gen_pipeline = transformers.pipeline(
20
  "text-generation",
21
+ # model = "ruslanmv/Medical-Llama3-v2",
22
+ model="ContactDoctor/Bio-Medical-Llama-3-2-1B-CoT-012025",
23
  torch_dtype=torch.bfloat16,
24
  device_map="auto"
25
  )