khalednabawi11 commited on
Commit
d3d58e4
·
verified ·
1 Parent(s): 8c3b05e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -236,9 +236,9 @@ QDRANT_URL = os.getenv("QDRANT_URL", "https://12efeef2-9f10-4402-9deb-f070977ddf
236
  QDRANT_API_KEY = os.getenv("QDRANT_API_KEY", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.Jb39rYQW2rSE9RdXrjdzKY6T1RF44XjdQzCvzFkjat4")
237
 
238
  # Load model and tokenizer
239
- model_name = "FreedomIntelligence/Apollo-7B"
240
  # model_name = "emilyalsentzer/Bio_ClinicalBERT"
241
- # model_name = "FreedomIntelligence/Apollo-2B"
242
 
243
  tokenizer = AutoTokenizer.from_pretrained(model_name)
244
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
236
  QDRANT_API_KEY = os.getenv("QDRANT_API_KEY", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.Jb39rYQW2rSE9RdXrjdzKY6T1RF44XjdQzCvzFkjat4")
237
 
238
  # Load model and tokenizer
239
+ # model_name = "FreedomIntelligence/Apollo-7B"
240
  # model_name = "emilyalsentzer/Bio_ClinicalBERT"
241
+ model_name = "FreedomIntelligence/Apollo-2B"
242
 
243
  tokenizer = AutoTokenizer.from_pretrained(model_name)
244
  model = AutoModelForCausalLM.from_pretrained(model_name)