Update app.py

app.py

@@ -63,10 +63,10 @@ def text2story(text):
         story = tokenizer.decode(outputs[0], skip_special_tokens=True)
         return story  
     except Exception as e:
-        # Fallback to deepseek Janus pro 7b if the advanced one fails
-        fallback_generator = AutoModelForCausalLM.from_pretrained("deepseek-ai/Janus-Pro-7B")
+        # Fallback to openai-community/gpt2 if the advanced one fails
+        fallback_generator = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
         fallback_prompt = f"Please generate the story for the following and target audience for kids : {text}"
-        tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/Janus-Pro-7B")
+        tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
         inputs = tokenizer(fallback_prompt, return_tensors="pt")
         fallback_story = fallback_generator.generate(
             inputs.input_ids,