alakxender commited on
Commit
495f10a
·
1 Parent(s): 322c280
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import spaces
2
  import gradio as gr
3
  from transformers import pipeline, GPT2TokenizerFast
4
 
@@ -8,7 +8,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)
10
 
11
- @spaces.GPU(duration=60)
12
  def generate_text(prompt, max_length, temperature):
13
  try:
14
  generated = generator(
 
1
+ #import spaces
2
  import gradio as gr
3
  from transformers import pipeline, GPT2TokenizerFast
4
 
 
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)
10
 
11
+ #@spaces.GPU(duration=60)
12
  def generate_text(prompt, max_length, temperature):
13
  try:
14
  generated = generator(