Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|