Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ import gradio
|
|
9 |
from PIL import Image
|
10 |
from timeit import default_timer as timer
|
11 |
from tensorflow import keras
|
|
|
12 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
13 |
from transformers import AutoModelWithLMHead, AutoTokenizer
|
14 |
import numpy as np
|
|
|
9 |
from PIL import Image
|
10 |
from timeit import default_timer as timer
|
11 |
from tensorflow import keras
|
12 |
+
import torch
|
13 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
14 |
from transformers import AutoModelWithLMHead, AutoTokenizer
|
15 |
import numpy as np
|