Update app.py

app.py

@@ -1,41 +1,34 @@
 import gradio as gr
-from transformers import pipeline
+from huggingface_hub import InferenceClient
 import os
 
-# 1. Load the model
-# NOTE: Ensure your model is PUBLIC on Hugging Face
-try:
-    pipe = pipeline("summarization", model="VoltIC/Automated-Text-Summarizer")
-    status = "Model loaded successfully!"
-except Exception as e:
-    pipe = None
-    status = f"Error: {str(e)}"
+# 1. Setup the Client
+# It will use the HF_TOKEN if you set it as a Secret, 
+# otherwise it works automatically if the model is Public.
+model_id = "VoltIC/Automated-Text-Summarizer"
+client = InferenceClient(model=model_id, token=os.getenv("HF_TOKEN"))
 
-def summarize(text):
-    if pipe is None:
-        return "Model not available. Check if repo is Public."
+def summarize_text(text):
     if not text.strip():
-        return "Please enter text."
+        return "Please enter text to summarize."
     
-    result = pipe(text, max_length=130, min_length=30, do_sample=False)
-    return result[0]['summary_text']
+    try:
+        # This calls the model remotely (No 1.63GB download needed!)
+        output = client.summarization(text)
+        return output
+    except Exception as e:
+        return f"Error: {str(e)}. Tip: Ensure model is Public or HF_TOKEN is set."
 
-# 2. Build UI using Blocks (Avoids the Interface IndexError)
+# 2. Simplified Interface to avoid the IndexError
 with gr.Blocks() as app:
-    gr.Markdown("# Aditya's Automated Text Summarizer")
-    gr.Markdown(f"**Status:** {status}")
+    gr.Markdown("# Aditya's Instant Summarizer")
+    gr.Markdown("Uses the HF Inference API to avoid large downloads.")
     
-    with gr.Row():
-        input_text = gr.Textbox(lines=10, label="Input Article", placeholder="Paste text here...")
+    input_box = gr.Textbox(lines=8, label="Input Article")
+    output_box = gr.Textbox(label="Summary")
+    submit_btn = gr.Button("Summarize")
     
-    with gr.Row():
-        submit_btn = gr.Button("Summarize")
-    
-    with gr.Row():
-        output_text = gr.Textbox(label="Summary")
-    
-    submit_btn.click(fn=summarize, inputs=input_text, outputs=output_text)
+    submit_btn.click(fn=summarize_text, inputs=input_box, outputs=output_box)
 
-# 3. Launch
 if __name__ == "__main__":
     app.launch()