acecalisto3 commited on
Commit
1c51e0f
·
verified ·
1 Parent(s): 2bf0c81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -60,8 +60,8 @@ def monitor_urls(storage_location, url1, url2, scrape_interval, content_type):
60
  options.add_argument("--no-sandbox")
61
  options.add_argument("--disable-dev-shm-usage")
62
 
63
- with webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options) as driver:
64
- try:
65
  while True:
66
  for i, url in enumerate(urls):
67
  try:
@@ -86,10 +86,8 @@ def monitor_urls(storage_location, url1, url2, scrape_interval, content_type):
86
  logging.error(f"Error accessing {url}: {e}")
87
 
88
  time.sleep(scrape_interval * 60) # Check every scrape_interval minutes
89
- except KeyboardInterrupt:
90
- logging.info("Monitoring stopped by user.")
91
- finally:
92
- driver.quit()
93
 
94
  # Define main function to handle user input
95
  def handle_input(storage_location, url1, url2, scrape_interval, content_type):
@@ -123,9 +121,11 @@ MAX_HISTORY = 125
123
 
124
  def format_prompt(message, history):
125
  prompt = "<s>"
126
- for user_prompt, bot_response in history:
127
- prompt += f"[INST] {user_prompt} [/INST]"
128
- prompt += f" {bot_response}</s> "
 
 
129
  prompt += f"[INST] {message} [/INST]"
130
  return prompt
131
 
@@ -196,7 +196,7 @@ demo = gr.Interface(
196
  gr.Textbox(label="Message"),
197
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
198
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
199
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
200
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
201
  gr.Textbox(value=default_file_path, label="Storage Location"),
202
  gr.Textbox(value="https://www.culver.k12.in.us/", label="URL 1"),
 
60
  options.add_argument("--no-sandbox")
61
  options.add_argument("--disable-dev-shm-usage")
62
 
63
+ try:
64
+ with webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options) as driver:
65
  while True:
66
  for i, url in enumerate(urls):
67
  try:
 
86
  logging.error(f"Error accessing {url}: {e}")
87
 
88
  time.sleep(scrape_interval * 60) # Check every scrape_interval minutes
89
+ except Exception as e:
90
+ logging.error(f"Error starting ChromeDriver: {e}")
 
 
91
 
92
  # Define main function to handle user input
93
  def handle_input(storage_location, url1, url2, scrape_interval, content_type):
 
121
 
122
  def format_prompt(message, history):
123
  prompt = "<s>"
124
+ for entry in history:
125
+ if isinstance(entry, tuple) and len(entry) == 2:
126
+ user_prompt, bot_response = entry
127
+ prompt += f"[INST] {user_prompt} [/INST]"
128
+ prompt += f" {bot_response}</s> "
129
  prompt += f"[INST] {message} [/INST]"
130
  return prompt
131
 
 
196
  gr.Textbox(label="Message"),
197
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
198
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
199
+ gr.Slider(minimum=0.1, maximum 4.0, value=0.7, step=0.1, label="Temperature"),
200
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
201
  gr.Textbox(value=default_file_path, label="Storage Location"),
202
  gr.Textbox(value="https://www.culver.k12.in.us/", label="URL 1"),