Ban text chat

app.py

@@ -44,7 +44,6 @@ if 'int4' in model_path:
     if device == 'mps':
         print('Error: running int4 model with bitsandbytes on Mac is not supported right now.')
         exit()
-    #model = AutoModel.from_pretrained(model_path, trust_remote_code=True, attn_implementation='sdpa')
     model = AutoModel.from_pretrained(model_path, trust_remote_code=True)
 else:
     if True: #args.multi_gpus:
@@ -282,6 +281,9 @@ def respond(_question, _chat_bot, _app_cfg, params_form):
     if files_cnts[1] + videos_cnt > 1 or (files_cnts[1] + videos_cnt == 1 and files_cnts[0] + images_cnt > 0):
         gr.Warning("Only supports single video file input right now!")
         return _question, _chat_bot, _app_cfg
+    if files_cnts[1] + videos_cnt + files_cnts[0] + images_cnt <= 0:
+        gr.Warning("Please chat with at least one image or video.")
+        return _question, _chat_bot, _app_cfg
 
     if params_form == 'Beam Search':
         params = {
@@ -349,9 +351,11 @@ def fewshot_add_demonstration(_image, _user_message, _assistant_message, _chat_b
 def fewshot_respond(_image, _user_message, _chat_bot, _app_cfg, params_form):
     user_message_contents = []
     _context = _app_cfg["ctx"].copy()
+    images_cnt = _app_cfg["images_cnt"]
     if _image:
         image = Image.open(_image).convert("RGB")
         user_message_contents += [encode_image(image)]
+        images_cnt += 1
     if _user_message:
         user_message_contents += [make_text(_user_message)]
     if user_message_contents:
@@ -373,6 +377,10 @@ def fewshot_respond(_image, _user_message, _chat_bot, _app_cfg, params_form):
             'repetition_penalty': 1.05,
             "max_new_tokens": 2048
         }
+
+    if images_cnt == 0:
+        gr.Warning("Please chat with at least one image or video.")
+        return _image, _user_message, '', _chat_bot, _app_cfg
     
     code, _answer, _, sts = chat("", _context, None, params)
 
@@ -391,6 +399,7 @@ def fewshot_respond(_image, _user_message, _chat_bot, _app_cfg, params_form):
     if code == 0:
         _app_cfg['ctx']=_context
         _app_cfg['sts']=sts
+        _app_cfg['images_cnt'] = images_cnt
     return None, '', '', _chat_bot, _app_cfg