Update app.py

app.py

@@ -20,8 +20,6 @@ st.title("OCR Image Text Extraction")
 uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"])
 
 if uploaded_file is not None:
-    # Open the uploaded image file
-    image = Image.open(uploaded_file)
     st.image(image, caption="Uploaded Image", use_column_width=True)
 
     # Add the spinner here while the model is processing
@@ -54,7 +52,7 @@ if uploaded_file is not None:
         inputs = inputs.to("cpu")
 
         # Inference: Generation of the output
-        generated_ids = MODEL.generate(**inputs, max_new_tokens=128)
+        generated_ids = MODEL.generate(**inputs, max_new_tokens=512)
         generated_ids_trimmed = [
             out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
         ]