Update app.py

app.py

@@ -63,20 +63,20 @@ def plot_sensor_data(df):
     plt.close(fig)
     return Image.open(fig_path)
 
-pipe = pipeline("visual-question-answering", model="ybelkada/blip2-opt-2.7b-fp16-sharded")     
+pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")     
 
 # Generate recommendations
 def analyze_data(image, plot_path):
     try:
-        # Ensure the image is a PIL object
+        # Ensure image is a PIL object
         if isinstance(image, np.ndarray):
             image = Image.fromarray(image)
 
         # Damage analysis
-        damage_output = pipe(image=image, question="Describe any visible damage in the image.")[0]['answer'] if image else "No image uploaded."
+        damage_output = pipe(inputs=image)[0]['generated_text'] if image else "No image uploaded."
 
         # Graph analysis
-        graph_analysis = pipe(image=plot_path, question="What anomalies or patterns can be observed?")[0]['answer']
+        graph_analysis = pipe(inputs=plot_path)[0]['generated_text']
 
         # Recommendations
         recommendations = f"Recommendations based on analysis:\n\n1. {graph_analysis}\n\n2. {damage_output}"
@@ -85,7 +85,6 @@ def analyze_data(image, plot_path):
     except Exception as e:
         return f"Error occurred: {str(e)}", None, None
 
-
 # Gradio UI
 with gr.Blocks(css=".output-text { font-family: 'Arial'; color: #222; font-size: 1rem; }") as app:
     gr.Markdown("# 🚗 Car Health Report Generation using Generative AI")