Update app.py

app.py

@@ -80,7 +80,7 @@ def start_scraping(storage_location, urls, scrape_interval, content_type, stop_s
         try:
             with webdriver.Chrome(service=Service(webdriver.ChromeDriverManager().install()), options=Options()) as driver:
                 driver.get(url)
-                time.sleep(2)  # Wait for the page to load
+                time.sleep(2 )  # Wait for the page to load
                 if content_type == "text":
                     initial_content = driver.page_source
                 elif content_type == "media":
@@ -104,6 +104,12 @@ def stop_scraping(stop_scraping_flag):
     stop_scraping_flag[0] = True
     return "Scraping stopped."
 
+# Define a function to update model config
+def update_model_config(model_name_input, gpu_layers_input):
+    # Example implementation of update_model_config
+    # You can replace this with the actual logic you need
+    return f"Model config updated with model: {model_name_input}, GPU layers: {gpu_layers_input}"
+
 # Define the Gradio interface
 def create_interface():
     with gr.Blocks() as demo:
@@ -128,10 +134,10 @@ def create_interface():
 
         stop_scraping_flag = [False]
         start_button.click(
-    fn=update_model_config,
-    inputs=[model_name_input, gpu_layers_input],
-    outputs="text"
-)
+            fn=update_model_config,
+            inputs=[model_name_input, gpu_layers_input],
+            outputs="text"
+        )
         stop_button.click(stop_scraping, inputs=[stop_scraping_flag], outputs=[csv_output])
         message.submit(chat_interface, inputs=[message, chat_history, system_message, max_tokens, temperature, top_p], outputs=[chat_history, response_box])