Update app.py

app.py

@@ -28,7 +28,7 @@ def reverse(sentence: str) -> str:
     return " ".join(sentence.split()[::-1])
 
 
-# ------------- Vocab class (must match training) -------------
+# ------------- Vocab class (same as training) -------------
 class Vocab:
     def __init__(self):
         self.word2idx = {'<PAD>': 0, '<SOS>': 1, '<EOS>': 2, '<UNK>': 3}
@@ -77,7 +77,7 @@ EOS_IDX = vocab.word2idx["<EOS>"]
 UNK_IDX = vocab.word2idx["<UNK>"]
 
 
-# ------------- model definitions (EXACTLY as in notebook) -------------
+# ------------- model definitions (same as notebook) -------------
 class Encoder(nn.Module):
     def __init__(self):
         super().__init__()
@@ -178,7 +178,7 @@ ckpt = torch.load(MODEL_FILE, map_location="cpu")
 model.load_state_dict(ckpt["model"])
 model.eval()
 
-print("✅ Model and vocab loaded. Chatbot ready to serve 🚀")
+print("Model and vocab loaded. Chatbot ready to serve ")
 
 
 # ------------- beam search (beam_generate_v2 from notebook) -------------
@@ -266,7 +266,7 @@ def generate_reply(user_text: str) -> str:
     src = torch.tensor([ids], dtype=torch.long, device=DEVICE)
     reply = beam_generate_v2(src, beam=5, max_len=50)
     if not reply.strip():
-        return "I don't know."
+        return "I'm a chatbot trained on Ubuntu Linux support conversations, so I may not understand this question."
     return reply
 
 
@@ -277,7 +277,7 @@ def respond(message, history):
 
 demo = gr.ChatInterface(
     fn=respond,
-    title="Ubuntu Chatbot (Seq2Seq + GRU + Attention)",
+    title="Ubuntu Chatbot (Seq2Seq + GRU + Attention) Developed by Group E",
     description="A generative chatbot trained on Ubuntu dialogue pairs (seq2seq with attention)."
 )