Spaces:

khalednabawi11
/

Medical-Chatbot-Apollo

Sleeping

khalednabawi11 commited on May 13

Commit

088ae0b

verified ·

1 Parent(s): 81a6746

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -176,8 +176,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextGenerationPipe
 import torch
 # Load model and tokenizer
-# model_name = "FreedomIntelligence/Apollo-7B"
-model_name = "emilyalsentzer/Bio_ClinicalBERT"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)

 import torch
 # Load model and tokenizer
+model_name = "FreedomIntelligence/Apollo-7B"
+# model_name = "emilyalsentzer/Bio_ClinicalBERT"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)