Update app.py

app.py

@@ -105,16 +105,15 @@ def _parse_text(text):
 
 def save_image(image_file, upload_dir):
     Path(upload_dir).mkdir(parents=True, exist_ok=True)
-    filename = secrets.token_hex(10) + Path(image_file.name).suffix
-    file_path = Path(upload_dir) / filename
-    if isinstance(image_file, gr.NamedString):
-        shutil.copy(image_file.name, file_path)
-    else:
+    if image_file is not None:
+        filename = secrets.token_hex(10) + Path(image_file.name).suffix
+        file_path = Path(upload_dir) / filename
         with open(file_path, "wb") as f:
             f.write(image_file.read())
-    return str(file_path)
+        return str(file_path)
+    else:
+        return None
 
-    
 def add_file(history, task_history, file):
     if file is None:
         return history, task_history
@@ -136,14 +135,11 @@ def _launch_demo(args, model, tokenizer):
             query = [{'image': chat_query[0]}]
         else:
             query = [{'text': _parse_text(chat_query)}]
-
         inputs = tokenizer.from_list_format(query)
         tokenized_inputs = tokenizer(inputs, return_tensors='pt')
         tokenized_inputs = tokenized_inputs.to(model.device)
-
         pred = model.generate(**tokenized_inputs)
         response = tokenizer.decode(pred.cpu()[0], skip_special_tokens=False)
-
         if 'image' in query[0]:
             image = tokenizer.draw_bbox_on_latest_picture(response)
             if image is not None: