Aadityaramrame commited on
Commit
8d2a2d7
·
verified ·
1 Parent(s): 8aa6871

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -2,9 +2,16 @@ from fastapi import FastAPI, Request, HTTPException
2
  from fastapi.middleware.cors import CORSMiddleware
3
  from tensorflow.keras.models import load_model
4
  import numpy as np
 
 
5
 
6
- # ---------- Load the Trained Model ----------
7
- MODEL_PATH = "cancer_classifier.h5"
 
 
 
 
 
8
  model = load_model(MODEL_PATH)
9
 
10
  # ---------- Initialize FastAPI ----------
 
2
  from fastapi.middleware.cors import CORSMiddleware
3
  from tensorflow.keras.models import load_model
4
  import numpy as np
5
+ import tensorflow as tf
6
+ from huggingface_hub import hf_hub_download
7
 
8
+ # 🔹 Download model file from your Hugging Face repo
9
+ MODEL_PATH = hf_hub_download(
10
+ repo_id="aadityaramrame/blood-cell-cancer-detector", # 👈 your repo
11
+ filename="cancer_classifier.h5"
12
+ )
13
+
14
+ # 🔹 Load model
15
  model = load_model(MODEL_PATH)
16
 
17
  # ---------- Initialize FastAPI ----------