nehalelkaref commited on
Commit
d30a365
·
1 Parent(s): 76d20e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,9 +5,9 @@ from huggingface_hub import Repository
5
  def greet(name):
6
  return "Hello " + name + "!!"
7
 
8
- args= dict({'transformer_model_name':'nehalelkaref/plain_span',
9
- 'device':'cuda'})
10
  iface = gr.Interface(fn=greet, inputs="text", outputs="text")
11
- model = SpanNet(args)
12
 
13
  # iface.launch()
 
5
  def greet(name):
6
  return "Hello " + name + "!!"
7
 
8
+ # args= dict({'transformer_model_name':'nehalelkaref/plain_span',
9
+ # 'device':'cuda'})
10
  iface = gr.Interface(fn=greet, inputs="text", outputs="text")
11
+ model = SpanNet(transformer_model_name= 'nehalelkaref/plain_span',device='cuda')
12
 
13
  # iface.launch()