Update app.py

app.py

@@ -2,33 +2,19 @@ import gradio as gr
 import time
 import os
 import sys
+import re
 import json
 import base64
 import tqdm
 import base64
 from datetime import datetime
 import subprocess
-import time
-from http import HTTPStatus
-from urllib3.exceptions import HTTPError
+
 from pathlib import Path
 import numpy as np
 from PIL import Image
-import os
-import numpy as np
-from urllib3.exceptions import HTTPError
-
 from argparse import ArgumentParser
-from pathlib import Path
-
-import copy
-import gradio as gr
-import os
-import re
-import secrets
 import tempfile
-import requests
-from http import HTTPStatus
 
 
 # Function to encode the image
@@ -60,58 +46,8 @@ openai_api_key = "EMPTY"
 openai_api_base = "http://47.117.17.202:9999/v1"
 
 
-client = OpenAI(
-    api_key=openai_api_key,
-    base_url=openai_api_base,
-)
-
-
-
-def run_example(image, text_input=None):
-
-    image_path = array_to_image_path(image)
-    content = [
-        {
-            "type": "image_url",
-            "image_url": {
-                "url": f"data:image/jpeg;base64,{encode_image(image_path)}"
-            }
-        }
-    ]
-
-    # 可选文本内容
-    if text_input is not None and text_input.strip():
-        content.append(
-            {
-                "type": "text",
-                "text": text_input
-            }
-        )
-
-    messages = [
-        {
-            "role": "user",
-            "content": content
-        }
-    ]
-    
-    response = client.chat.completions.create(
-                model="Qwen2_5VL",
-                messages=messages,
-                extra_body={},
-                extra_headers={
-                    "apikey": "empty"
-                },
-                stream=False,
-                temperature=0.7,
-                top_p=1.0,
-            )
-  
-    return response.choices[0].message.content
 
-
-
-REVISION = 'v1.0.4'
+REVISION = 'v0.0.1'
 BOX_TAG_PATTERN = r"<box>([\s\S]*?)</box>"
 PUNCTUATION = "！？。＂＃＄％＆＇（）＊＋，－／：；＜＝＞＠［＼］＾＿｀｛｜｝～｟｠｢｣､、〃》「」『』【】〔〕〖〗〘〙〚〛〜〝〞〟〰〾〿–—‘’‛“”„‟…‧﹏."
 
@@ -203,11 +139,14 @@ def _launch_demo(args):
                             "url": f"data:image/jpeg;base64,{encode_image(q[0])}"
                         }
                     })        
-            else:
+            elif q and q.strip(): 
                 content.append({"type": "text", 'text': q})
                 messages.append({'role': 'user', 'content': content})
                 messages.append({'role': 'assistant', 'content': [{"type": "text", 'text': a}]})
                 content = []
+            else:
+                continue
+                
         messages.pop()
         responses = client.chat.completions.create(
                 model="Qwen2_5VL",
@@ -286,9 +225,9 @@ def _launch_demo(args):
         return [], []
 
     with gr.Blocks() as demo:
-        gr.Markdown("""<center><font size=3> Qwen2.5-VL-32B-Instruct Demo </center>""")
+        gr.Markdown("""<center><font size=3> Infinity-7B Demo </center>""")
 
-        chatbot = gr.Chatbot(label='Qwen2.5-VL-32B-Instruct', elem_classes="control-height", height=500)
+        chatbot = gr.Chatbot(label='Infinity-7B', elem_classes="control-height", height=500)
         query = gr.Textbox(lines=2, label='Input')
         task_history = gr.State([])