dsv5

app.py

@@ -10,11 +10,13 @@ import spaces
 import math
 import os
 
-from transformers import Qwen2_5_VLForConditionalGeneration
+# from transformers import Qwen2_5_VLForConditionalGeneration
 from qwen_vl_utils import process_vision_info, fetch_image
 
 
-pretrained_model_name_or_path=os.environ.get("MODEL", "amrn/testmodel2")
+# pretrained_model_name_or_path=os.environ.get("MODEL", "amrn/testmodel2")
+pretrained_model_name_or_path=os.environ.get("MODEL", "amrn/dsv5mx3")
+
 auth_token = os.environ.get("HF_TOKEN") or True
 DEFAULT_PROMPT = "Find abnormalities and support devices."
 
@@ -117,7 +119,7 @@ with gr.Blocks() as demo:
 
     send_btn = gr.Button("Send", variant="primary", render=False)
     textbox = gr.Textbox(show_label=False, placeholder="Enter your text here and press ENTER", render=False, submit_btn="Send")
-    chatbot = gr.Chatbot(type="messages", label="AI", render_markdown=True, sanitize_html=False, allow_tags=True, height=800, container=False, show_share_button=False)
+    # chatbot = gr.Chatbot(type="messages", label="AI", render_markdown=True, sanitize_html=False, allow_tags=True, height=800, container=False, show_share_button=False)
 
     with gr.Row():
         with gr.Column(scale=4):
@@ -141,8 +143,8 @@ with gr.Blocks() as demo:
             chat_interface = gr.ChatInterface(fn=model_inference,
                 # title='title', description='description',
                 type="messages",
-                chatbot=chatbot,
-                textbox=textbox,
+                chatbot=gr.Chatbot(type="messages", label="AI", render_markdown=True, sanitize_html=False, allow_tags=True, height=800, container=False, show_share_button=False)
+,                textbox=textbox,
                 additional_inputs=image_input,
                 multimodal=False,
                 )