Update app.py

app.py

@@ -4,13 +4,13 @@ from PIL import Image
 import easyocr
 from transformers import pipeline
 
-# Create a text2text-generation pipeline with the "t5-small" model
+# Create a text2text-generation pipeline with the "t5-base" model
 pipe = pipeline("text2text-generation", model="t5-base")
 
 # Initialize the EasyOCR reader for text extraction from images
 ocr_reader = easyocr.Reader(['en'])
 
-st.title("Text Explanation Model")
+st.title("Text-to-Story Generator")
 
 uploaded_file = st.file_uploader("Upload an image:")
 
@@ -25,12 +25,13 @@ if uploaded_file is not None:
     st.markdown(extracted_text)
 
     if extracted_text:
-        # Use the pipeline to generate a concise explanation
-        explanation = pipe(extracted_text, max_length=150, do_sample=True)[0]["generated_text"]
-        st.markdown("**Explanation (5-6 lines):**")
-        st.markdown(explanation)
+        # 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 = pipe(story_prompt, max_length=300, do_sample=True)[0]["generated_text"]
+        st.markdown("**Story:**")
+        st.markdown(story)
     else:
         st.warning("No text extracted from the image.")
 
 else:
-    st.markdown("Please upload an image to extract text and generate an explanation.")
+    st.markdown("Please upload an image to extract text and generate a story.")