Update app.py

app.py

@@ -27,24 +27,19 @@ from PIL import Image
 from io import BytesIO
 import torchvision.transforms as T
 import requests
-from pathlib import Path
+import model_loader
 
-MODEL_URL = "https://www.dropbox.com/s/04suaimdpru76h3/ArtLine_920.pkl?dl=1"
-MODEL_PATH = "ArtLine_920.pkl"
+app = FastAPI()
 
-# Download the model file
-response = requests.get(MODEL_URL)
-if response.status_code == 200:
-    with open(MODEL_PATH, 'wb') as f:
-        f.write(response.content)
-else:
-    print("Failed to download the model")
+# Download the model if not already downloaded
+MODEL_URL = "https://www.dropbox.com/s/04suaimdpru76h3/ArtLine_920.pkl?dl=1"
+MODEL_FILENAME = "ArtLine_920.pkl"
+if not os.path.exists(MODEL_FILENAME):
+    model_loader.download_model(MODEL_URL, MODEL_FILENAME)
 
-# Load the model using FastAI
-path = Path(".")
-learn = load_learner(path, MODEL_PATH)
+# Load the model
+learn = model_loader.load_model(MODEL_FILENAME)
 
-app = FastAPI()
 
 class FeatureLoss(nn.Module):
     def __init__(self, m_feat, layer_ids, layer_wgts):