Update app.py

app.py

@@ -15,7 +15,7 @@ with open("textgen_tokenizer.pkl", "rb") as f:
     textgen_tokenizer = pickle.load(f)
 
 # === Settings ===
-CLASS_LABELS = ['label_0', 'label_1', 'label_2']  # Replace with actual labels
+CLASS_LABELS = ['Science', 'Math', 'History']  # Replace with actual labels
 MAX_LEN_CLASSIFIER = 100
 MAX_LEN_TEXTGEN = 50
 
@@ -29,7 +29,7 @@ def classify_text(text):
 
 
 # === Text Generation with Top-k Sampling ===
-def generate_text(seed_text, next_words=30, k=10, temperature=0.9):
+def generate_text(seed_text, next_words=15, k=10, temperature=0.9):
     recent_words = set()
     for _ in range(next_words):
         token_list = textgen_tokenizer.texts_to_sequences([seed_text])[0]