wasmdashai commited on
Commit
8c06717
·
verified ·
1 Parent(s): 47a7353

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import gradio as gr
2
  GK=0
3
  from transformers import AutoTokenizer
@@ -590,7 +592,7 @@ wandb.login(key= "782b6a6e82bbb5a5348de0d3c7d40d1e76351e79")
590
  wandb.init(project= 'AZ',config = training_args.to_dict())
591
 
592
 
593
-
594
  def greet(text,id):
595
  global GK
596
  b=int(id)
 
1
+ import spaces
2
+
3
  import gradio as gr
4
  GK=0
5
  from transformers import AutoTokenizer
 
592
  wandb.init(project= 'AZ',config = training_args.to_dict())
593
 
594
 
595
+ @spaces.GPU
596
  def greet(text,id):
597
  global GK
598
  b=int(id)