Upload app.py with huggingface_hub

app.py

@@ -28,7 +28,7 @@ def answer_question(image, question):
     encoding = processor(pil_image, question, return_tensors="pt")
     outputs = model(**encoding)
     idx = outputs.logits.argmax(-1).item()
-    return processor.decode(idx)
+    return model.config.id2label[idx]
 
 css = """
 .gradio-container {
@@ -54,7 +54,7 @@ css = """
 .footer { text-align: center; padding: 1.5rem; color: #64748b; font-size: 0.875rem; }
 """
 
-with gr.Blocks(theme=theme, css=css, title="Visual Q&A") as demo:
+with gr.Blocks(title="Visual Q&A") as demo:
     with gr.Row():
         with gr.Column():
             gr.HTML("""
@@ -123,4 +123,4 @@ with gr.Blocks(theme=theme, css=css, title="Visual Q&A") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(theme=theme)
+    demo.launch(theme=theme, css=css)