Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,11 +18,7 @@ class Medication(BaseModel):
|
|
| 18 |
total_daily_dose_mg: Optional[float] = Field(default=None, description="The total daily dose in milligrams.")
|
| 19 |
|
| 20 |
# Load your model locally via llama-cpp
|
| 21 |
-
model = LlamaCpp(
|
| 22 |
-
model_path="/path/to/cmcmaster/drug_parsing_Llama-3.2-1B-Instruct-Q5_K_S-GGUF.gguf", # Change this path
|
| 23 |
-
temperature=0.0,
|
| 24 |
-
max_tokens=512
|
| 25 |
-
)
|
| 26 |
|
| 27 |
sampler = samplers.greedy()
|
| 28 |
|
|
|
|
| 18 |
total_daily_dose_mg: Optional[float] = Field(default=None, description="The total daily dose in milligrams.")
|
| 19 |
|
| 20 |
# Load your model locally via llama-cpp
|
| 21 |
+
model = LlamaCpp("cmcmaster/drug_parsing_Llama-3.2-1B-Instruct-Q5_K_S-GGUF", "drug_parsing_llama-3.2-1b-instruct-q5_k_s.gguf")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
sampler = samplers.greedy()
|
| 24 |
|