input length check added

app.py

@@ -7,7 +7,10 @@ import os
 API_URL = "https://api-inference.huggingface.co/models/Rahmat82/t5-small-finetuned-summarization-xsum"
 SECRET_KEY = os.environ.get("summarizer")
 
+# asynchronous function
 async def summarize(text, retries=3):
+    if len(text)) < 20:
+        return "Error: Input should be at least few sentences long!"
     headers = {"Authorization": f"Bearer {SECRET_KEY}"}
     data = {
         "inputs": text,
@@ -24,18 +27,21 @@ async def summarize(text, retries=3):
         for attempt in range(retries):
             try:
                 async with session.post(API_URL, headers=headers, json=data) as response:
-                    response.raise_for_status()  # Raise exception for non-200 status codes
+                    # raise exception for non-200 status codes
+                    response.raise_for_status()  
                     result = await response.json()
                     return result[0]["summary_text"]
             except aiohttp.ClientResponseError as e:
-                # Retry only if the error is 503 (Service Unavailable)
+                # retry only if the error is 503 (Service Unavailable)
                 if e.status == 503 and attempt < retries - 1:
                     print(f"Retry attempt {attempt+1} for 503 error.")
-                    await asyncio.sleep(1)  # Add a short delay before retrying
+                    # add a short delay before retrying
+                    await asyncio.sleep(1)  
                     continue
                 else:
                     return f"Error: {e.status}, message='{e.message}'"
 
+# define gradio interface
 iface = gr.Interface(
     theme=gr.themes.Soft(),
     fn=summarize,