File size: 511 Bytes
98bebfc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import subprocess
import gradio as gr
from PIL import Image
import requests
import torch
import uuid
import shutil
import json
import yaml
from transformers import AutoProcessor, AutoModelForCausalLM
from tqdm import tqdm
import os
if __name__ == "__main__":
# url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg?download=true"
image = Image.open("examples/cartoon_dinosaur.png").convert("RGB")
breakpoint()
run_captioning(image)
|