Hugging Face's logo

Spaces:
ai-forever
/
mGPT-armenian
Runtime error

App Files Community
ai-forever commited on
Commit
64a5201
•
1 Parent(s): 031192b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
  import gradio as gr
3
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
4
- from googletrans import Translator
5
 
6
- translator = Translator()
7
  tokenizer = GPT2Tokenizer.from_pretrained("sberbank-ai/mGPT-armenian")
8
  model = GPT2LMHeadModel.from_pretrained("sberbank-ai/mGPT-armenian")
9
  #model.cuda()
 
1
  import torch
2
  import gradio as gr
3
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
4
+ #from googletrans import Translator
5
 
6
+ #translator = Translator()
7
  tokenizer = GPT2Tokenizer.from_pretrained("sberbank-ai/mGPT-armenian")
8
  model = GPT2LMHeadModel.from_pretrained("sberbank-ai/mGPT-armenian")
9
  #model.cuda()