Update app.py

app.py

@@ -29,7 +29,7 @@ def summarize(req: SummarizeRequest):
     models = {
         "bart": "kUrrooYuki/bart-large-cnn-text-summarize",
         "t5": "kUrrooYuki/t5-base-text-summarize",
-        "led": "kUrrooYuki/led-base-16384-text-summarize"
+        "led": "kUrrooYuki/led-base-text-summarize"
     }
     
     if req.model_type not in models:
@@ -60,6 +60,16 @@ def summarize(req: SummarizeRequest):
         
         print("-> [3/3] Komputasi selesai! Menerjemahkan token ke teks...", flush=True)
         summary_result = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+
+        summary_result = summary_result.strip()
+
+        if summary_result.lower().startswith("a's "):
+            summary_result = summary_result[4:].strip()
+        elif summary_result.lower().startswith("a "):
+            summary_result = summary_result[2:].strip()
+
+        if len(summary_result) > 0:
+            summary_result = summary_result[0].upper() + summary_result[1:]
         
         del tokenizer
         del model