Update app.py

app.py

@@ -37,8 +37,8 @@ def generate_response(message, history):
     
     with torch.no_grad():
         text = model.generate(**inputs, max_new_tokens=256, do_sample=False)
-    start = text.find('<|object_ref_end|>') + len('<|object_ref_end|>')
-    end = text.find('<|box_end|>')
+    start = text[0].find('<|object_ref_end|>') + len('<|object_ref_end|>')
+    end = text[0].find('<|box_end|>')
     result = text[start:end]
     return tokenizer.decode(result, skip_special_tokens=False)
 chatbot = gr.ChatInterface(