Update src/app.py
Browse files- src/app.py +2 -1
src/app.py
CHANGED
|
@@ -4,12 +4,13 @@ import tensorflow as tf
|
|
| 4 |
from tensorflow.keras.models import load_model
|
| 5 |
from PIL import Image
|
| 6 |
import numpy as np
|
|
|
|
| 7 |
|
| 8 |
from huggingface_hub import hf_hub_download
|
| 9 |
|
| 10 |
st.title("lung cancer detection")
|
| 11 |
st.write("Upload an image of a lung X-ray to detect lung cancer.")
|
| 12 |
-
|
| 13 |
# Download model from Hugging Face model hub
|
| 14 |
model_path = hf_hub_download(
|
| 15 |
repo_id="lp128396/lung_cancer_model", # 👈 replace with your actual username and repo
|
|
|
|
| 4 |
from tensorflow.keras.models import load_model
|
| 5 |
from PIL import Image
|
| 6 |
import numpy as np
|
| 7 |
+
import os
|
| 8 |
|
| 9 |
from huggingface_hub import hf_hub_download
|
| 10 |
|
| 11 |
st.title("lung cancer detection")
|
| 12 |
st.write("Upload an image of a lung X-ray to detect lung cancer.")
|
| 13 |
+
os.environ["HF_HOME"] = "/tmp"
|
| 14 |
# Download model from Hugging Face model hub
|
| 15 |
model_path = hf_hub_download(
|
| 16 |
repo_id="lp128396/lung_cancer_model", # 👈 replace with your actual username and repo
|