Commit ·
78f97ac
1
Parent(s): 4d430a8
Fix: Update the path and fix conflicts
Browse files- Dockerfile +1 -0
- inference.py +2 -1
Dockerfile
CHANGED
|
@@ -2,6 +2,7 @@ FROM python:3.10-slim
|
|
| 2 |
|
| 3 |
WORKDIR /app
|
| 4 |
|
|
|
|
| 5 |
# Install system dependencies for RDKit
|
| 6 |
RUN apt-get update && apt-get install -y \
|
| 7 |
libxrender1 \
|
|
|
|
| 2 |
|
| 3 |
WORKDIR /app
|
| 4 |
|
| 5 |
+
|
| 6 |
# Install system dependencies for RDKit
|
| 7 |
RUN apt-get update && apt-get install -y \
|
| 8 |
libxrender1 \
|
inference.py
CHANGED
|
@@ -9,7 +9,8 @@ import logging
|
|
| 9 |
# =========================
|
| 10 |
# PATHS (HF Spaces SAFE)
|
| 11 |
# =========================
|
| 12 |
-
|
|
|
|
| 13 |
MODEL_DIR = BASE_DIR / "models" / "biogpt-lora-finetuned"
|
| 14 |
QML_MODEL_PATH = BASE_DIR / "models" / "qml_model.pth"
|
| 15 |
|
|
|
|
| 9 |
# =========================
|
| 10 |
# PATHS (HF Spaces SAFE)
|
| 11 |
# =========================
|
| 12 |
+
|
| 13 |
+
BASE_DIR = Path(__file__).resolve().parent # root folder
|
| 14 |
MODEL_DIR = BASE_DIR / "models" / "biogpt-lora-finetuned"
|
| 15 |
QML_MODEL_PATH = BASE_DIR / "models" / "qml_model.pth"
|
| 16 |
|