ChaficSalloum's picture
Create app.py
f0b49ea verified
raw
history blame contribute delete
620 Bytes
import gradio as gr
from transformers import CLIPProcessor, CLIPModel
from PIL import Image
# Load FashionCLIP (Patrickjohncyh’s version)
model = CLIPModel.from_pretrained("patrickjohncyh/fashion-clip")
processor = CLIPProcessor.from_pretrained("patrickjohncyh/fashion-clip")
def analyze_fashion(image):
inputs = processor(images=image, return_tensors="pt")
outputs = model.get_image_features(**inputs)
# Convert tensor to list for JSON response
return {"features": outputs.detach().numpy().tolist()}
demo = gr.Interface(fn=analyze_fashion, inputs=gr.Image(type="pil"), outputs="json")
demo.launch()