runaksh commited on
Commit
6c761bc
·
1 Parent(s): acc5a03

Update app.py

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