Update app.py

app.py

@@ -104,6 +104,8 @@ examples = [
 def test(image, question):
     image = image_processor(images=image.convert('RGB'), return_tensors='pt')['pixel_values']
     image_features = visual_encoder.encode_images(image.unsqueeze(0))
+    emb_mixer.set_image_embeddings(image_features)
+    model.w["emb.weight"] = emb_mixer.get_input_embeddings()
     image_ids = [i for i in range(emb_mixer.image_start_index, emb_mixer.image_start_index + len(image_features))]
     input_text = generate_prompt(question)
     for output in generate(input_text, image_ids):
@@ -117,4 +119,4 @@ demo = gr.Interface(fn=test,
                     description="VisualRWKV-v5.0")
 
 demo.queue(concurrency_count=1, max_size=10)
-demo.launch(share=False)
+demo.launch(share=False, enable_queue=True)