Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import io
 from PIL import Image
 import easyocr
-from transformers import pipeline
+from transformers import pipeline, set_seed
 
 # Create a text2text-generation pipeline with the "t5-base" model
 pipe = pipeline("text2text-generation", model="t5-base")
@@ -26,7 +26,8 @@ if uploaded_file is not None:
 
     if extracted_text:
         # Use the pipeline to generate a story based on the extracted text
-        story_prompt = f"Create a story based on the following text: '{extracted_text}'"
+        story_prompt = f"Once upon a time, in a world where '{extracted_text}',"
+        set_seed(42)  # Set a seed for reproducibility
         story = pipe(story_prompt, max_length=300, do_sample=True)[0]["generated_text"]
         st.markdown("**Story:**")
         st.markdown(story)