Update app.py

app.py

@@ -37,10 +37,13 @@ def generate_response(message, history):
     
     with torch.no_grad():
         text = model.generate(**inputs, max_new_tokens=256, do_sample=False)
-    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)
+    start = full_output.find('<|object_ref_end|>') + len('<|object_ref_end|>')
+    end = full_output.find('<|box_end|>')
+    if end == -1:  # اگر <|box_end|> پیدا نشد
+        result = full_output[start:]
+    else:
+        result = full_output[start:end]
+    return result
 chatbot = gr.ChatInterface(
     fn=generate_response,
     title="Simple ZeroGPU Chatbot",