Update app.py

app.py

@@ -4,8 +4,8 @@ from PIL import Image
 import easyocr
 from transformers import pipeline
 
-# Create a text2text-generation pipeline with the "google/flan-t5-base" model
-pipe = pipeline("text2text-generation", model="google/flan-t5-base")
+# Create a text2text-generation pipeline with the "t5-small" model
+pipe = pipeline("text2text-generation", model="t5-small")
 
 # Initialize the EasyOCR reader for text extraction from images
 ocr_reader = easyocr.Reader(['en'])
@@ -26,7 +26,7 @@ if uploaded_file is not None:
 
     if extracted_text:
         # Use the pipeline to generate a concise explanation
-        explanation = pipe(extracted_text, max_length=30, do_sample=True)[0]["generated_text"]
+        explanation = pipe(extracted_text, max_length=150, do_sample=True)[0]["generated_text"]
         st.markdown("**Explanation (5-6 lines):**")
         st.markdown(explanation)
     else: