Spaces:

slliac
/

5240-indiv-assignment

Running

slliac commited on Mar 2

Commit

299d8b5

verified ·

1 Parent(s): 258d5ac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -69,7 +69,7 @@ def text2story(text):
         fallback_prompt = f"{text}"
         tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
         inputs = tokenizer(fallback_prompt, return_tensors="pt")
-        fallback_story = model.generate(
             inputs.input_ids,
             max_new_tokens=250,
             temperature=0.7,

         fallback_prompt = f"{text}"
         tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
         inputs = tokenizer(fallback_prompt, return_tensors="pt")
+        fallback_story = fallback_generator.generate(
             inputs.input_ids,
             max_new_tokens=250,
             temperature=0.7,