Update app.py

app.py

@@ -4,37 +4,32 @@ import requests
 import gradio as gr
 from transformers import pipeline
 
-class WikipediaBot:
-    SEARCH_TEMPLATE = "https://en.wikipedia.org/w/api.php?action=opensearch&search=%s&limit=1&namespace=0&format=json"
-    CONTENT_TEMPLATE = "https://en.wikipedia.org/w/api.php?format=json&action=query&prop=extracts&exintro&explaintext&redirects=1&titles=%s"
-
-    def __init__(self):
-        self.summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
-
-    @spaces.GPU
-    def search_wikipedia(self, query):
-        query = urllib.parse.quote_plus(query)
-        data = requests.get(self.SEARCH_TEMPLATE % query).json()
-        if data and data[1]:
-            page = urllib.parse.quote_plus(data[1][0])
-            content = requests.get(self.CONTENT_TEMPLATE % page).json()
-            content = list(content["query"]["pages"].values())[0]["extract"]
-            summary = self.summarizer(content, max_length=150, min_length=50, do_sample=False)[0]['summary_text']
-            source = data[3][0]
-            return summary, source
-        else:
-            return "No results found.", ""
-
-    def chatbot_response(self, message, history):
-        summary, source = self.search_wikipedia(message)
-        response = f"Here's a summary of the top Wikipedia result:\n\n{summary}"
-        if source:
-            response += f"\n\nSource: {source}"
-        history.append((message, response))
-        return history
-
-def create_chatbot():
-    return WikipediaBot()
+SEARCH_TEMPLATE = "https://en.wikipedia.org/w/api.php?action=opensearch&search=%s&limit=1&namespace=0&format=json"
+CONTENT_TEMPLATE = "https://en.wikipedia.org/w/api.php?format=json&action=query&prop=extracts&exintro&explaintext&redirects=1&titles=%s"
+
+summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
+
+@spaces.GPU
+def search_wikipedia(query):
+    query = urllib.parse.quote_plus(query)
+    data = requests.get(SEARCH_TEMPLATE % query).json()
+    if data and data[1]:
+        page = urllib.parse.quote_plus(data[1][0])
+        content = requests.get(CONTENT_TEMPLATE % page).json()
+        content = list(content["query"]["pages"].values())[0]["extract"]
+        summary = summarizer(content, max_length=150, min_length=50, do_sample=False)[0]['summary_text']
+        source = data[3][0]
+        return summary, source
+    else:
+        return "No results found.", ""
+
+def chatbot_response(message, history):
+    summary, source = search_wikipedia(message)
+    response = f"Here's a summary of the top Wikipedia result:\n\n{summary}"
+    if source:
+        response += f"\n\nSource: {source}"
+    history.append((message, response))
+    return history
 
 demo = gr.Blocks()
 
@@ -46,9 +41,7 @@ with demo:
     msg = gr.Textbox(label="Enter your query")
     clear = gr.Button("Clear")
 
-    bot = create_chatbot()
-
-    msg.submit(bot.chatbot_response, [msg, chatbot], chatbot).then(
+    msg.submit(chatbot_response, [msg, chatbot], chatbot).then(
         lambda: gr.update(value=""), None, [msg], queue=False
     )
     clear.click(lambda: None, None, chatbot, queue=False)