Update app.py

app.py

@@ -7,6 +7,13 @@ from io import BytesIO
 from bs4 import BeautifulSoup
 import pandas as pd
 import base64
+import os
+
+def download_model(model_url, model_path):
+    if not os.path.exists(model_path):
+        response = requests.get(model_url)
+        with open(model_path, 'wb') as f:
+            f.write(response.content)
 
 def load_model(model_path):
     interpreter = tf.lite.Interpreter(model_path=model_path)
@@ -50,7 +57,10 @@ def main():
     st.title("Shopping Intent Classification - SEO by DEJAN")
     st.write("Enter a URL to fetch and classify all images on the page. Javascript-based website scraping currently unsupported.")
 
+    model_url = "https://huggingface.co/dejanseo/shopping-intent/resolve/main/model.tflite"
     model_path = "model.tflite"
+    download_model(model_url, model_path)
+
     url = st.text_input("Enter URL")
 
     if url: