ai: Ready to drink.

README.md

@@ -8,5 +8,20 @@ app_file: jarvis.py
 pinned: true
 short_description: Inspired by Iron Man movies.
 models:
+  - deepseek-ai/DeepSeek-V3-0324
+  - deepseek-ai/DeepSeek-R1
+  - deepseek-ai/DeepSeek-R1-Distill-Qwen-32B
+  - deepseek-ai/DeepSeek-R1-Distill-Llama-70B
+  - google/gemma-3-1b-it
+  - google/gemma-3-4b-it
+  - google/gemma-3-27b-it
+  - meta-llama/Llama-3.1-8B-Instruct
+  - meta-llama/Llama-3.2-3B-Instruct
+  - meta-llama/Llama-3.3-70B-Instruct
+  - meta-llama/Llama-4-Maverick-17B-128E-Instruct
+  - meta-llama/Llama-4-Scout-17B-16E-Instruct
+  - Qwen/Qwen2.5-VL-3B-Instruct
+  - Qwen/Qwen2.5-VL-32B-Instruct
+  - Qwen/Qwen2.5-VL-72B-Instruct
   - Qwen/QwQ-32B
 ---

jarvis.py

@@ -59,8 +59,7 @@ def marked_item(item, marked, attempts):
     if attempts[item] >= 3:
         def remove_fail():
             marked.discard(item)
-            if item in attempts:
-                del attempts[item]
+            attempts.pop(item, None)
         threading.Timer(3600, remove_fail).start()
 
 class SessionWithID(requests.Session):
@@ -137,7 +136,7 @@ async def chat_with_model_async(history, user_input, selected_model_display, ses
     selected_model = get_model_key(selected_model_display)
     model_config = MODEL_CONFIG.get(selected_model, DEFAULT_CONFIG)
     messages = [{"role": "user", "content": user} for user, _ in history] + [{"role": "assistant", "content": assistant} for _, assistant in history if assistant]
-    if INTERNAL_TRAINING_DATA:
+    if INTERNAL_TRAINING_DATA and MODEL_CHOICES and selected_model_display == MODEL_CHOICES[0]:
         messages.insert(0, {"role": "system", "content": INTERNAL_TRAINING_DATA})
     messages.append({"role": "user", "content": user_input})
     global ACTIVE_CANDIDATE
@@ -196,5 +195,8 @@ with gr.Blocks(fill_height=True, fill_width=True, title=AI_TYPES["AI_TYPE_4"], h
     chatbot = gr.Chatbot(label=AI_TYPES["AI_TYPE_1"], show_copy_button=True, scale=1, elem_id=AI_TYPES["AI_TYPE_2"])
     with gr.Row():
         msg = gr.MultimodalTextbox(show_label=False, placeholder=RESPONSES["RESPONSE_5"], interactive=True, file_count="single", file_types=ALLOWED_EXTENSIONS)
+    with gr.Accordion(AI_TYPES["AI_TYPE_6"], open=False):
+        model_dropdown = gr.Dropdown(show_label=False, choices=MODEL_CHOICES, value=MODEL_CHOICES[0])
+    model_dropdown.change(fn=change_model, inputs=[model_dropdown], outputs=[user_history, user_session, selected_model], show_progress="full")
     msg.submit(fn=respond_async, inputs=[msg, user_history, selected_model, user_session], outputs=[chatbot, msg, user_session], api_name=INTERNAL_AI_GET_SERVER)
 jarvis.launch(max_file_size="1mb")