Update app.py

app.py

@@ -0,0 +1,34 @@
+import gradio as gr
+from transformers import AutoProcessor, AutoModelForImageTextToText
+from PIL import Image
+
+# Load model and processor
+processor = AutoProcessor.from_pretrained("guneetsk99/finance_qwen_VL_7B")
+model = AutoModelForImageTextToText.from_pretrained("guneetsk99/finance_qwen_VL_7B")
+
+def predict(input_img):
+    # Preprocess the image
+    inputs = processor(images=input_img, return_tensors="pt")
+    
+    # Generate predictions using the model
+    outputs = model.generate(**inputs)
+    
+    # Decode the generated text
+    generated_text = processor.decode(outputs[0], skip_special_tokens=True)
+    
+    # Return the input image and the generated text
+    return input_img, {"Prediction": generated_text}
+
+# Create the Gradio interface
+gradio_app = gr.Interface(
+    predict,
+    inputs=gr.Image(label="Upload Image", source="upload", type="pil"),
+    outputs=[
+        gr.Image(label="Uploaded Image"),
+        gr.Label(label="Generated Text"),
+    ],
+    title="Image to Text Model",
+)
+
+if __name__ == "__main__":
+    gradio_app.launch()