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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ 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
 
 
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