test

gradio_app.py

@@ -5,11 +5,13 @@ import io
 import pandas as pd
 import spaces
 
+from generate import stream_file
+
+
 @spaces.GPU(duration=120)
 def stream_output(filename: str):
     if filename.endswith(".jsonl"):
         filename = filename[:-len(".jsonl")]
-    from generate import stream_file
     content = ""
     size=3
     start_time = time.time()
@@ -71,7 +73,7 @@ with gr.Blocks() as demo:
             with gr.Row():
                 file_content_comp = gr.Markdown()
 
-    generate_button.click(test, filename_comp, [dataframe_comp, file_content_comp, state_msg_comp])
+    generate_button.click(stream_output, filename_comp, [dataframe_comp, file_content_comp, state_msg_comp])
 
 
 demo.launch()