Update app.py

app.py

@@ -2,14 +2,6 @@ import re
 import gradio as gr
 import torch
 
-from transformers import (
-    AutoTokenizer,
-    AutoModelForSeq2SeqLM
-)
-
-import re
-import gradio as gr
-import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 MODEL_NAME = "google/flan-t5-small"
@@ -17,11 +9,14 @@ MAX_INPUT_TOKENS = 512
 MAX_OUTPUT_TOKENS = 300
 
 print("Loading model...")
+
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 model.eval()
+
 print("Model loaded.")
 
+
 # -----------------------------
 # Utilities
 # -----------------------------
@@ -68,7 +63,7 @@ def build_prompt(mode: str, text: str) -> str:
 
 
 # -----------------------------
-# Core inference
+# Inference
 # -----------------------------
 
 @torch.inference_mode()
@@ -92,7 +87,7 @@ def generate_text(prompt: str) -> str:
 
 
 # -----------------------------
-# Public API function
+# API function
 # -----------------------------
 
 def process(text: str, mode: str) -> str:
@@ -105,7 +100,7 @@ def process(text: str, mode: str) -> str:
 
 
 # -----------------------------
-# Gradio UI (API-first friendly)
+# Gradio app
 # -----------------------------
 
 demo = gr.Interface(
@@ -120,14 +115,9 @@ demo = gr.Interface(
     ],
     outputs=gr.Textbox(lines=8, label="Output"),
     title="Free Text Rewrite & Generation Service",
-    description=(
-        "Fast, lightweight AI service using flan-t5-small.\n"
-        "Supports rewriting, normalization, short definitions, and article generation."
-    ),
-    allow_flagging="never"
+    description="Lightweight rewrite, normalization, definitions, and articles using flan-t5-small."
 )
 
-# IMPORTANT: no queue (avoids async / stuck Space issues)
 demo.queue(False)
 
 demo.launch(