import gradio as gr
import tensorflow as tf
import numpy as np

from transformers import AutoModelForSequenceClassification

# model = AutoModelForSequenceClassification.from_pretrained("Alesteba/deep_model_07")

from huggingface_hub import from_pretrained_keras
model = from_pretrained_keras("Alesteba/deep_model_07")

def predict(prompt):
    return model.predict(prompt)[0]
    
gr.Interface(fn=predict, inputs="text", outputs="text").launch()