Spaces:

talinm2
/

llava-1.5-dlai

Runtime error

talinm2 commited on Nov 20, 2024

Commit

4ebe676

verified ·

1 Parent(s): 9ed82e1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,4 +23,4 @@ def inference(text, image):
     output = model.generate(**inputs, max_new_tokens=1024)
     return processor.decode(output[0], skip_special_tokens=True).split("ASSISTANT:")[-1]
-gr.Interface(fn=inference, inputs=[gr.Text(), gr.Image()], outputs=gr.Text()).launch()

     output = model.generate(**inputs, max_new_tokens=1024)
     return processor.decode(output[0], skip_special_tokens=True).split("ASSISTANT:")[-1]
+gr.Interface(fn=inference, inputs=[gr.Text(), gr.Image()], outputs=gr.Text()).launch()