acecalisto3 commited on
Commit
14db2f5
·
verified ·
1 Parent(s): 857c0ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -127,6 +127,9 @@ def create_interface():
127
  start_button = gr.Button("Start Scraping")
128
  stop_button = gr.Button("Stop Scraping")
129
  csv_output = gr.Textbox(label="CSV Output", interactive=False)
 
 
 
130
 
131
  with gr.Column():
132
  chat_history = gr.Chatbot(label="Chat History")
@@ -134,9 +137,9 @@ def create_interface():
134
 
135
  stop_scraping_flag = [False]
136
  start_button.click(
137
- fn=update_model_config,
138
- inputs=[model_name_input, gpu_layers_input],
139
- outputs="text"
140
  )
141
  stop_button.click(stop_scraping, inputs=[stop_scraping_flag], outputs=[csv_output])
142
  message.submit(chat_interface, inputs=[message, chat_history, system_message, max_tokens, temperature, top_p], outputs=[chat_history, response_box])
 
127
  start_button = gr.Button("Start Scraping")
128
  stop_button = gr.Button("Stop Scraping")
129
  csv_output = gr.Textbox(label="CSV Output", interactive=False)
130
+
131
+ model_name_input = gr.Textbox(value="default_model", label="Model Name")
132
+ gpu_layers_input = gr.Slider(minimum=0, maximum=8, value=2, step=1, label="GPU Layers")
133
 
134
  with gr.Column():
135
  chat_history = gr.Chatbot(label="Chat History")
 
137
 
138
  stop_scraping_flag = [False]
139
  start_button.click(
140
+ fn=start_scraping,
141
+ inputs=[storage_location, urls, scrape_interval, content_type, stop_scraping_flag],
142
+ outputs=csv_output
143
  )
144
  stop_button.click(stop_scraping, inputs=[stop_scraping_flag], outputs=[csv_output])
145
  message.submit(chat_interface, inputs=[message, chat_history, system_message, max_tokens, temperature, top_p], outputs=[chat_history, response_box])