Spaces:

SaherMuhamed
/

intent-classifier-chatbot

Sleeping

SaherMuhamed commited on Jul 9, 2025

Commit

3479dfb

1 Parent(s): c6efb71

add the fine tuned BERT model with FAST API integrated in the Flask app

Files changed (8) hide show

Dockerfile CHANGED Viewed

@@ -17,4 +17,4 @@ EXPOSE 8000
 # Hugging Face Spaces expects the app to run on 0.0.0.0:8000
 ENV FLASK_APP=src.main
-CMD ["flask", "run", "--host=0.0.0.0", "--port=8000", "--no-debugger", "--no-reload"]

 # Hugging Face Spaces expects the app to run on 0.0.0.0:8000
 ENV FLASK_APP=src.main
+CMD ["flask", "run", "--host=0.0.0.0", "--port=5000", "--no-debugger", "--no-reload"]

model/api/__pycache__/api.cpython-39.pyc ADDED Viewed

Binary file (1.72 kB). View file

model/api/api.py CHANGED Viewed

@@ -13,7 +13,7 @@ BASE_DIR = os.path.dirname(BASE_DIR)
 MODEL_DIR = os.path.join(BASE_DIR, "intent_classifier_model")
 TOKENIZER_DIR = os.path.join(BASE_DIR, "intent_classifier_tokenizer")
-# Ensure model and tokenizer directories exist
 if not os.path.isdir(MODEL_DIR):
     raise FileNotFoundError(f"Model directory not found: {MODEL_DIR}")
 if not os.path.isdir(TOKENIZER_DIR):
@@ -23,17 +23,10 @@ if not os.path.isdir(TOKENIZER_DIR):
 model = BertForSequenceClassification.from_pretrained(MODEL_DIR, local_files_only=True)
 tokenizer = BertTokenizer.from_pretrained(TOKENIZER_DIR, local_files_only=True)
-# Static intent label mapping (example for first few intents, fill in all 151 as needed)
-INTENT_LABELS = [
-    "restaurant_reviews", "nutrition_info", "account_blocked", "oil_change_how", "time", "weather", "redeem_rewards",
-    "interest_rate", "gas_type", "accept_reservations", "smart_home", "user_name", "report_lost_card", "repeat",
-    "whisper_mode", "what_are_your_hobbies", "order", "jump_start", "schedule_meeting", "meeting_schedule",
-    "freeze_account", "what_song", "meaning_of_life", "restaurant_reservation", "traffic", "pay_bill", "transit_delay",
-    "card_declined", "balance", "tell_joke", "last_maintenance", "exchange_rate", "uber", "car_rental", "credit_limit",
-    "oos", # ... continue for all 151 intents ...
-]
-def int2str(idx):
-    return INTENT_LABELS[idx] if 0 <= idx < len(INTENT_LABELS) else "unknown"
 class Query(BaseModel):
     text: str

 MODEL_DIR = os.path.join(BASE_DIR, "intent_classifier_model")
 TOKENIZER_DIR = os.path.join(BASE_DIR, "intent_classifier_tokenizer")
+# Ensure model and tokenizer directories exist
 if not os.path.isdir(MODEL_DIR):
     raise FileNotFoundError(f"Model directory not found: {MODEL_DIR}")
 if not os.path.isdir(TOKENIZER_DIR):
 model = BertForSequenceClassification.from_pretrained(MODEL_DIR, local_files_only=True)
 tokenizer = BertTokenizer.from_pretrained(TOKENIZER_DIR, local_files_only=True)
+# Load intent label mapping
+from datasets import load_dataset
+dataset = load_dataset("clinc_oos", "small")
+int2str = dataset["train"].features["intent"].int2str
 class Query(BaseModel):
     text: str

model/api/start_server.py CHANGED Viewed

@@ -2,4 +2,3 @@ import uvicorn
 if __name__ == "__main__":
     uvicorn.run("api:app", host="0.0.0.0", port=8000, reload=True)


2
3	if __name__ == "__main__":
4	uvicorn.run("api:app", host="0.0.0.0", port=8000, reload=True)

model/api/test.py CHANGED Viewed

@@ -1,9 +1,8 @@
 import requests
 url = "http://localhost:8000/predict"
-data = {"text": "I want to set an alarm for 7 AM tomorrow."}
 response = requests.post(url, json=data)
 print("Status code:", response.status_code)
 print("Response:", response.json())

 import requests
 url = "http://localhost:8000/predict"
+data = {"text": "I need to add 100$ to my bank account."}
 response = requests.post(url, json=data)
 print("Status code:", response.status_code)
 print("Response:", response.json())

src/{main.py → app.py} RENAMED Viewed

@@ -31,4 +31,4 @@ def index():
     return render_template("index.html", prediction=prediction, user_text=user_text)
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=8080, debug=False)

     return render_template("index.html", prediction=prediction, user_text=user_text)
 if __name__ == "__main__":
+    app.run(debug=True)

src/fastapi_server.py CHANGED Viewed

@@ -15,4 +15,3 @@ def predict(req: PredictRequest):
     else:
         intent = "unknown"
     return {"intent": intent}

     else:
         intent = "unknown"
     return {"intent": intent}

src/templates/index.html CHANGED Viewed

@@ -24,12 +24,6 @@
             color: #2d6cdf;
             margin-bottom: 18px;
         }
-        h2 {
-            text-align: center;
-            color: #2d6cdf;
-            margin-bottom: 18px;
-            font-size: 1.5em;
-        }
         label {
             font-weight: 500;
             margin-bottom: 8px;
@@ -97,7 +91,6 @@
 <body>
     <div class="container">
         <h1>Intent Classifier Chatbot</h1>
-        <h2>Predict User Intent</h2>
         <div class="info">
             Enter a message below and click <b>Predict Intent</b> to see what the AI thinks your intent is.<br>
             <span style="color:#2d6cdf;">Try: <i>"Set an alarm for 7am"</i> or <i>"Transfer money to John"</i></span>

             color: #2d6cdf;
             margin-bottom: 18px;
         }
         label {
             font-weight: 500;
             margin-bottom: 8px;
 <body>
     <div class="container">
         <h1>Intent Classifier Chatbot</h1>
         <div class="info">
             Enter a message below and click <b>Predict Intent</b> to see what the AI thinks your intent is.<br>
             <span style="color:#2d6cdf;">Try: <i>"Set an alarm for 7am"</i> or <i>"Transfer money to John"</i></span>