Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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)
|