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)
|