Update app.py

app.py

@@ -19,14 +19,10 @@ def generate_image_caption(image):
     Returns:
         str: The generated caption.
     """
-    # Convert image to RGB format
     image = image.convert("RGB")
-    
-    # Preprocess the image and generate a caption
     inputs = processor(images=image, return_tensors="pt")
     outputs = model.generate(**inputs)
     caption = processor.decode(outputs[0], skip_special_tokens=True)
-    
     return caption
 
 def generate_keywords(caption):
@@ -40,7 +36,7 @@ def generate_keywords(caption):
         list: A list of single-word keywords.
     """
     words = caption.split()
-    keywords = list(set(words))[:50]  # Take unique words and limit to 50
+    keywords = list(set(words))[:50]
     return keywords
 
 def process_images(image_files):
@@ -55,7 +51,7 @@ def process_images(image_files):
     """
     metadata = []
     temp_dir = tempfile.mkdtemp()
-    
+
     for image_file in image_files:
         filename = os.path.basename(image_file.name)
         image = Image.open(image_file)
@@ -68,7 +64,7 @@ def process_images(image_files):
                 'title': title,
                 'keywords': keywords
             })
-    
+
     # Create CSV file
     csv_file_path = os.path.join(temp_dir, 'images_metadata.csv')
     with open(csv_file_path, mode='w', newline='', encoding='utf-8') as file:
@@ -79,14 +75,14 @@ def process_images(image_files):
             title = data['title']
             keywords = ','.join(data['keywords'])
             writer.writerow([filename, title, keywords])
-    
+
     return [Image.open(image_file.name) for image_file in image_files], csv_file_path
 
 # Define Gradio interface
 iface = gr.Interface(
     fn=process_images,
     inputs=[
-        gr.File(label="Upload Image Files", type="file", multiple=True)
+        gr.Files(label="Upload Image Files")  # Use gr.Files for multiple file uploads
     ],
     outputs=[
         gr.Gallery(label="Processed Images"),