Change main function name

app.py

@@ -3,13 +3,13 @@ import gradio as gr
 
 model = pipeline('summarization', model='facebook/bart-large-cnn')
 
-def predict(text):
+def summarize(text):
     summary = model(text)[0]['summary_text']
     return summary
 
 with gr.Blocks() as ui:
     textbox = gr.Textbox(placeholder='Enter text to summarize', label='input')
-    gr.Interface(fn=predict, inputs=textbox, outputs='text')
+    gr.Interface(fn=summarize, inputs=textbox, outputs='text')
 
 if __name__ == '__main__':
     ui.launch()

test_app.py

@@ -1,11 +1,11 @@
-from app import predict
+from app import summarize
 import wikipedia
 wikipedia.set_lang("it")
 import warnings
 warnings.filterwarnings("ignore")
 
 
-def test_predict():
+def test_summarize():
     text = wikipedia.summary('Sardegna', sentences=10)
-    summary = predict(text)
+    summary = summarize(text)
     assert len(text) > len(summary)