app.py

import gradio as gr
from network import SpanNet
from huggingface_hub import Repository

def greet(name):
    return "Hello " + name + "!!"

# args= dict({'transformer_model_name':'nehalelkaref/plain_span',
#             'device':'cuda'})
iface = gr.Interface(fn=greet, inputs="text", outputs="text")
model = SpanNet(transformer_model_name= 'nehalelkaref/plain_span',device='cuda')

# iface.launch()