Hugging Face's logo

Spaces:
alakxender
/
dv-gpt2-textgen
Running

App Files Community
alakxender commited on
Commit
ea7f2a6
·
1 Parent(s): b6e2b47

fx

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +4 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import spaces
2
  import gradio as gr
3
  from transformers import pipeline, GPT2TokenizerFast
4
 
5
- #model_id = "alakxender/dv-wiki-gpt2"
6
- model_id = "alakxender/dv-articles-gpt2"
7
  tokenizer = GPT2TokenizerFast.from_pretrained(model_id, model_max_length=128)
8
 
9
  generator = pipeline("text-generation", model=model_id, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, eos_token_id=tokenizer.eos_token_id)
 
2
  import gradio as gr
3
  from transformers import pipeline, GPT2TokenizerFast
4
 
5
+ model_id = "alakxender/dv-wiki-gpt2"
6
+ #model_id = "alakxender/dv-articles-gpt2"
7
  tokenizer = GPT2TokenizerFast.from_pretrained(model_id, model_max_length=128)
8
 
9
  generator = pipeline("text-generation", model=model_id, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, eos_token_id=tokenizer.eos_token_id)
requirements.txt CHANGED
@@ -1,3 +1,5 @@
1
  transformers
2
- tensorflow
3
- tf-keras
 
 
 
1
  transformers
2
+ tf-keras
3
+ torch
4
+ torchvision
5
+ torchaudio