slliac commited on
Commit
258d5ac
·
verified ·
1 Parent(s): d9851a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def text2story(text):
68
  fallback_generator = AutoModel.from_pretrained("openai-community/gpt2")
69
  fallback_prompt = f"{text}"
70
  tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
71
- inputs = tokenizer(prompt, return_tensors="pt")
72
  fallback_story = model.generate(
73
  inputs.input_ids,
74
  max_new_tokens=250,
 
68
  fallback_generator = AutoModel.from_pretrained("openai-community/gpt2")
69
  fallback_prompt = f"{text}"
70
  tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
71
+ inputs = tokenizer(fallback_prompt, return_tensors="pt")
72
  fallback_story = model.generate(
73
  inputs.input_ids,
74
  max_new_tokens=250,