import gradio as gr
from transformers import AutoModel, AutoTokenizer

# Load the model and tokenizer
model_name = "jazzmacedo/fruits-and-vegetables-detector-36"
model = AutoModel.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

# Define a prediction function
def predict(input_text):
    inputs = tokenizer(input_text, return_tensors="pt")
    outputs = model(**inputs)
    return str(outputs)

# Create a Gradio interface
iface = gr.Interface(
    fn=predict,
    inputs="text",
    outputs="text",
    title="Fruits and Vegetables Detector",
    description="Enter text to get predictions."
)

iface.launch()