Models_space / app.py
lcyyyy's picture
Update app.py
42ac98b verified
raw
history blame contribute delete
601 Bytes
from transformers import pipeline
# Step 1: Load the pre-trained model
image_to_text_pipeline = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
# Step 2: Get the uploaded image file path
uploaded_files = get_uploaded_files()
if not uploaded_files:
print("Please upload an image file.")
else:
image_path = uploaded_files[0] # Assuming only one image is uploaded
# Step 3: Test the model by providing the uploaded image
generated_text = image_to_text_pipeline(image_path)
# Step 4: Print the generated text
print("Generated Text:", generated_text)