testmodel / app.py
Pavan99's picture
update
9e4cb18 verified
#import gradio as gr
#def greet(name):
# return "Hello " + name + "!!"
#gr.load("models/fathyshalaby/emailclassifier").launch()
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="fathyshalaby/emailclassifier")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("fathyshalaby/emailclassifier")
model = AutoModelForSequenceClassification.from_pretrained("fathyshalaby/emailclassifier")
#iface = gr.Interface(fn=greet, inputs="text", outputs="text")
#iface.launch()