Update app.py

app.py

@@ -2,9 +2,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 import torch
 
-title = "AI ChatBot"
-description = "A State-of-the-Art Large-scale Pretrained Response generation model (DialoGPT)"
-examples = [["How are you?"]]
+title = "GPT-2 JTE Chatbot"
+description = "GPT-2 Menjadi Chatbot"
+examples = [["Bagaimana cara mengisi KRS?"]]
 
 tokenizer = AutoTokenizer.from_pretrained("betajuned/GPT-2_Kombinasi4")
 model = AutoModelForCausalLM.from_pretrained("betajuned/GPT-2_Kombinasi4")
@@ -17,7 +17,7 @@ def predict(input, history=[]):
     bot_input_ids = torch.cat([torch.tensor(history), new_user_input_ids], dim=-1) if history else new_user_input_ids
 
     # Generate a response
-    chat_history_ids = model.generate(bot_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
+    chat_history_ids = model.generate(bot_input_ids, max_length=50, pad_token_id=tokenizer.eos_token_id)
 
     # Convert the tokens to text
     response = tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)