Update app.py

app.py

@@ -4,18 +4,14 @@ import re
 
 app = Flask(__name__)
 
-# Load the NER pipeline
+# Use a PUBLICLY AVAILABLE model that works on free tier
 ner_pipeline = pipeline(
     "ner",
-    model="VoiceLab/bert-base-uncased-finetuned-ner-fashion",
+    model="dslim/bert-base-NER",  # Verified public model
     aggregation_strategy="simple"
 )
 
 def extract_entities(query):
-    # Get entities from model
-    entities = ner_pipeline(query)
-    
-    # Initialize result dictionary
     result = {
         "Brand": None,
         "Category": None,
@@ -23,19 +19,24 @@ def extract_entities(query):
         "Price": None
     }
     
+    # Extract entities
+    entities = ner_pipeline(query)
+    
     # Process entities
     for entity in entities:
-        label = entity["entity_group"]
-        value = entity["word"].strip()
-        
-        if label == "Brand" and not result["Brand"]:
-            result["Brand"] = value
-        elif label == "Product Type" and not result["Category"]:
-            result["Category"] = value
-        elif label == "Gender" and not result["Gender"]:
-            result["Gender"] = value
+        if entity["entity_group"] == "ORG":  # Organizations are likely brands
+            result["Brand"] = entity["word"]
+    
+    # Add keyword-based extraction for other fields
+    query_lower = query.lower()
+    if "perfume" in query_lower or "cologne" in query_lower:
+        result["Category"] = "Perfume"
+    if "men" in query_lower:
+        result["Gender"] = "Men"
+    elif "women" in query_lower:
+        result["Gender"] = "Women"
     
-    # Extract price using regex (since model doesn't handle prices)
+    # Price extraction
     price_match = re.search(r"under (\d+)\s*AED", query, re.IGNORECASE)
     if price_match:
         result["Price"] = f"Under {price_match.group(1)} AED"
@@ -51,4 +52,4 @@ def index():
     return render_template("index.html", result=None)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=5000, debug=True)
+    app.run(debug=True, host="0.0.0.0", port=7860)