Update app.py

app.py

@@ -1,6 +1,6 @@
+import gradio as gr
 from fastai.text.all import *
 from blurr.text.modeling.all import *
-import gradio as gr
 
 # Load the exported model
 learn = load_learner('article_highlights.pkl')
@@ -10,10 +10,14 @@ def generate_summary(article_text):
     summary = learn.blurr_generate_summary(article_text)
     return summary
 
-def main():
-    article_text = input()
-    summary = generate_summary(article_text)
-    print(summary)
-
-if __name__ == "__main__":
-    main()
+iface = gr.Interface(
+    fn=generate_summary,
+    inputs="text",
+    outputs="text",
+    title="Article Summarizer",
+    description="Enter an article and get a summary.",
+    examples=[
+        ["Text of an article goes here..."]
+    ]
+)
+iface.launch()