Create app.py

app.py

@@ -0,0 +1,48 @@
+from flask import Flask, request, jsonify, render_template
+from transformers import BlipProcessor, BlipForConditionalGeneration
+from PIL import Image
+import torch
+import os
+
+app = Flask(__name__)
+
+# Load the processor and model
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
+model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
+
+@app.route('/')
+def home():
+    return render_template('index.html')
+
+@app.route('/upload', methods=['POST'])
+def upload():
+    if 'file' not in request.files:
+        return jsonify({"error": "No file uploaded"}), 400
+
+    file = request.files['file']
+    
+    if file.filename == '':
+        return jsonify({"error": "No selected file"}), 400
+
+    # Save the file to a temporary location
+    image_path = os.path.join('uploads', file.filename)
+    file.save(image_path)
+
+    # Load and process the image
+    image = Image.open(image_path)
+    inputs = processor(image, return_tensors="pt")
+
+    # Generate caption
+    with torch.no_grad():
+        out = model.generate(**inputs)
+
+    # Decode the output
+    caption = processor.decode(out[0], skip_special_tokens=True)
+
+    # Clean up the uploaded file
+    os.remove(image_path)
+
+    return jsonify({"caption": caption})
+
+if __name__ == '__main__':
+    app.run(debug=True)