set default models

app.py

@@ -59,6 +59,12 @@ def inference(image_input,
     if model_state is None or tokenizer_state is None:
         chatbot.append(("System", "Please select a model to start the conversation."))
         return chatbot, state, ""   
+
+    # Check for empty or invalid user message
+    if not user_message or user_message.strip() == '' or user_message.lower() == 'system':
+        chatbot.append(("System", "Please enter a valid message to continue the conversation."))
+        return chatbot, state, ""
+
     
     model = model_state
     tokenizer = tokenizer_state
@@ -126,22 +132,27 @@ def regenerate_response(chatbot,
     
     model = model_state
     tokenizer = tokenizer_state
-    
-    
+        
     # Check if there is a previous user message
     if chatbot is None or len(chatbot) == 0:
         chatbot = []
         chatbot.append(("System", "Nothing to regenerate. Please start a conversation first."))
         return chatbot, state, 
     
-    # Check if there is a previous user message
-    if state is None or len(state) == 0:
-        chatbot.append(("System", "Nothing to regenerate. Please start a conversation first."))
-        return chatbot, state
+    # # Check if there is a previous user message
+    # if state is None or len(state) == 0:
+    #     chatbot.append(("System", "Nothing to regenerate. Please start a conversation first."))
+    #     return chatbot, state
     
     # Get the last user message
     last_user_message, _ = chatbot[-1]
     
+    # Check for empty or invalid last user message
+    if not last_user_message or last_user_message.strip() == '' or last_user_message.lower() == 'system':
+        chatbot.append(("System", "Cannot regenerate response for an empty or invalid message."))
+        return chatbot, state
+    
+    
     state = state[:-1]  # Remove last assistant's response from history
    
     if len(state) == 0 or not state:
@@ -191,7 +202,8 @@ with gr.Blocks() as demo:
     with gr.Row():
         model_dropdown = gr.Dropdown(
             choices=list(model_paths.keys()),
-            label="Select Model"
+            label="Select Model",
+            value="H2OVL-Mississippi-2B"
         )
     
     # When the model selection changes, load the new model
@@ -299,4 +311,4 @@ with gr.Blocks() as demo:
         label = "examples",
     )
     
-demo.launch()
+demo.launch(share=True)