working

app.py

@@ -12,7 +12,7 @@ HF_TOKEN = os.environ['HF_TOKEN']
 
 def summarize(model_name, article):
     API_URL = f"https://api-inference.huggingface.co/models/{model_name}"
-    headers = {"Authorization": "Bearer {HF_TOKEN}"}
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
     payload = {"inputs": article}
     response = requests.post(API_URL, headers=headers, json=payload)
@@ -83,25 +83,22 @@ def scrape_article(url):
     except Exception as e:
         return "Error scraping article: " + str(e), ""
 
-# Using Gradio Blocks with improved layout and styling
 with gr.Blocks() as demo:
-    gr.Markdown("# News Summary App", elem_id="header")
+    gr.Markdown("# News Summary App")
     gr.Markdown("Enter a news text and get its summary, or load a random article.")
     
     with gr.Row():
         with gr.Column():
-            with gr.Row():
-                load_dataset_article_button = gr.Button("Load Random Article from Val Dataset")
-                load_news_article_button = gr.Button("Pull Random News Article from NewsAPI")
+            load_dataset_article_button = gr.Button("Load Random Article from Dataset")
+            load_news_article_button = gr.Button("Load News Article")
             article_title = gr.Label()  # Component to display the article title
-            input_text = gr.Textbox(lines=10, label="Input Text")
+            input_text = gr.Textbox(lines=10, label="Input Text", placeholder="Enter article text or load a random article...")
         with gr.Column():
-            with gr.Row():
-                summarize_button = gr.Button("Summarize")
             model_name = gr.Dropdown(label="Model Name", choices=["liamvbetts/bart-news-summary-v1", "liamvbetts/bart-base-cnn-v1", "liamvbetts/bart-large-cnn-v2", "liamvbetts/bart-large-cnn-v4"], value="liamvbetts/bart-news-summary-v1")
-            output_text = gr.Textbox(label="Summary")
+            summarize_button = gr.Button("Summarize")
+            output_text = gr.Textbox(label="Summary", placeholder="Summary will appear here...")
 
-    load_dataset_article_button.click(fn=load_article, inputs=[], outputs=input_text)
+    load_dataset_article_button.click(fn=load_article, inputs=[], outputs=[input_text, article_title])
     load_news_article_button.click(fn=get_news_article, inputs=[], outputs=[input_text, article_title])
     summarize_button.click(fn=summarize, inputs=[model_name, input_text], outputs=output_text)
 

requirements.txt

@@ -1,7 +1,3 @@
 gradio
-#transformers
 datasets
-evaluate
-accelerate
-torch
 beautifulsoup4