remove exception

app.py

@@ -59,20 +59,16 @@ def vicuna_respond(tab_name, message, chat_history):
     formatted_prompt = f'''Generate the output only for the assistant. Please output any {tab_name} in the following sentence one per line without any additional text: {message}'''
     print('Vicuna - Prompt + Context:')
     print(formatted_prompt)
-    try:
-        input_ids = vicuna_tokenizer.encode(formatted_prompt, return_tensors="pt")
-        output_ids = vicuna_model.generate(input_ids, do_sample=True, max_length=100, num_beams=5, no_repeat_ngram_size=2)
-        bot_message = vicuna_tokenizer.decode(output_ids[0], skip_special_tokens=True)
-        # Remove formatted prompt from bot_message
-        bot_message = bot_message.replace(formatted_prompt, '')
-        print(bot_message)
+    input_ids = vicuna_tokenizer.encode(formatted_prompt, return_tensors="pt")
+    output_ids = vicuna_model.generate(input_ids, do_sample=True, max_length=100, num_beams=5, no_repeat_ngram_size=2)
+    bot_message = vicuna_tokenizer.decode(output_ids[0], skip_special_tokens=True)
+    # Remove formatted prompt from bot_message
+    bot_message = bot_message.replace(formatted_prompt, '')
+    print(bot_message)
         
-        chat_history.append((formatted_prompt, bot_message))
-        time.sleep(2)
-        return tab_name, "", chat_history
-    except Exception as e:
-        print(f"Error in vicuna_respond: {str(e)}")
-        return tab_name, "", chat_history
+    chat_history.append((formatted_prompt, bot_message))
+    time.sleep(2)
+    return tab_name, "", chat_history
 
 def llama_respond(tab_name, message, chat_history):
     formatted_prompt = f'''Generate the output only for the assistant. Please output any {tab_name} in the following sentence one per line without any additional text: {message}'''