Update app.py

app.py

@@ -41,8 +41,8 @@ def generate_captions_from_image(image):
     if image.mode != "RGB":
         image = image.convert("RGB")
     
-    # Resize image for faster processing
-    image = image.resize((640, 640))
+    # Resize image for faster processing (use smaller resolution to speed up inference)
+    image = image.resize((320, 320))  # Reduced size for faster processing
     
     # Preprocess the image and generate a caption
     inputs = processor(image, return_tensors="pt").to(device, torch.float16)
@@ -146,7 +146,6 @@ def upload_file_to_salesforce(file_path, filename, sf_connection, file_type):
     except Exception as e:
         return None, None, f"Error uploading {filename} to Salesforce: {str(e)}"
 
-# Function to generate the daily progress report (DPR), save as PDF, and upload to Salesforce
 # Function to generate the daily progress report (DPR), save as PDF, and upload to Salesforce
 def generate_dpr(files):
     dpr_text = []
@@ -249,7 +248,6 @@ def generate_dpr(files):
         dpr_output + f"\n\n{pdf_result}\n\nSalesforce Upload Status:\n{salesforce_result}",
         pdf_filepath
     )
-
 # Gradio interface for uploading multiple files, displaying DPR, and downloading PDF
 iface = gr.Interface(
     fn=generate_dpr,