tunisian-arabiz / app.py
RamiIbrahim's picture
5:11
60e6eca verified
raw
history blame
810 Bytes
import gradio as gr
from transformers import AutoModelForSequenceClassification, AutoTokenizer
# Load your model and tokenizer
model_name = "your-model-name"
model = AutoModelForSequenceClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
def predict(input_text):
# Tokenize and make prediction
inputs = tokenizer(input_text, return_tensors="pt", truncation=True, padding=True)
outputs = model(**inputs)
prediction = outputs.logits.argmax(-1).item()
return prediction
# Define Gradio interface
iface = gr.Interface(
fn=predict,
inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
outputs="label",
title="Your Model Name",
description="Enter some text to get a prediction."
)
# Launch the interface
iface.launch()