Spaces:
Sleeping
Sleeping
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) | |