Update app.py

app.py

@@ -6,6 +6,8 @@ from io import BytesIO
 import requests
 import torch.nn as nn
 import os
+import tempfile
+import shutil
 
 # Define the FeatureLoss class
 class FeatureLoss(nn.Module):
@@ -67,8 +69,10 @@ if not os.path.exists(model_file_path):
             f.write(response.content)
     st.success('Model downloaded successfully!')
 
-# Load the trained model
-learn = load_learner(model_file_path)
+# Create a temporary directory for the model
+with tempfile.TemporaryDirectory() as tmpdirname:
+    shutil.move(model_file_path, os.path.join(tmpdirname, 'export.pkl'))
+    learn = load_learner(tmpdirname)
 
 # Input for image URL or path
 image_path_or_url = st.text_input("Enter image path or URL", "")