Update app.py

app.py

@@ -3,10 +3,8 @@ import numpy as np
 from datasets import load_dataset
 import gradio as gr
 
-
-text_dataset = load_dataset("HuggingFaceFW/fineweb", name="sample-10BT", split="train", streaming=True, columns=['text'])
-
-def update(text_dataset):
+def update():
+    text_dataset = load_dataset("HuggingFaceFW/fineweb", name="sample-10BT", split="train", streaming=True, columns=['text'])
     bigram_vectorizer = CountVectorizer(ngram_range=(2, 2), analyzer="word")
     co_occurrences = bigram_vectorizer.fit_transform(doc['text'] for doc in text_dataset)
     print('Printing sparse matrix:')
@@ -20,6 +18,6 @@ def update(text_dataset):
 with gr.Blocks() as app:
     gr.Markdown("Click **Run** to start calculating.")
     btn = gr.Button("Run")
-    btn.click(fn=update, inputs=text_dataset)
+    btn.click(fn=update)
 
 app.launch()