Update app.py

app.py

@@ -26,8 +26,6 @@ TITLE = "<h1><center>VL-Chatbox</center></h1>"
 
 DESCRIPTION = "<h3><center>MODEL: " + MODEL_NAME + "</center></h3>"
 
-DEFAULT_SYSTEM = "You named Chatbox. You are a good assitant."
-
 CSS = """
 .duplicate-button {
   margin: auto !important;
@@ -50,9 +48,9 @@ eos_token_id=processor.tokenizer.eos_token_id
 
 
 @spaces.GPU(queue=False)
-def stream_chat(message, history: list, system: str, temperature: float, max_new_tokens: int):
+def stream_chat(message, history: list, temperature: float, max_new_tokens: int):
     print(message)
-    conversation = [{"role": "user", "content": system or DEFAULT_SYSTEM}]
+    conversation = []
     for prompt, answer in history:
         conversation.extend([{"role": "user", "content": f"<|image_1|>\n{prompt}"}, {"role": "assistant", "content": answer}])
     conversation.append({"role": "user", "content": message['text']})
@@ -60,6 +58,8 @@ def stream_chat(message, history: list, system: str, temperature: float, max_new
     if message["files"]:
         image = Image.open(message["files"][-1]).convert('RGB')
     else:
+        if len(history) == 0:
+            gr.Error("Please upload an image first.")
         image = None
         
     prompt = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
@@ -111,11 +111,6 @@ with gr.Blocks(css=CSS) as demo:
         fill_height=True,
         additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
         additional_inputs=[
-            gr.Text(
-                value="",
-                label="System",
-                render=False,
-            ),
             gr.Slider(
                 minimum=0,
                 maximum=1,