app.py

import streamlit as st
from transformers import AutoModelForImageClassification, AutoImageProcessor
from ultralytics import YOLO
from PIL import Image
import requests
from io import BytesIO
import time
import torch
import concurrent.futures


# Streamlit app UI
st.title("Clothing Classification Pipeline")

url = st.sidebar.text_input("Paste image URL here...")
if url:
    try:
        response = requests.get(url)
        if response.status_code == 200:
            image = Image.open(BytesIO(response.content))
            st.sidebar.image(image.resize((200, 200)), caption="Uploaded Image", use_column_width=False)
    except:
        st.write("Url Invalid...!")