Update app.py

app.py

@@ -10,13 +10,19 @@ def load_ocr_model():
 # Load text-generation model
 @st.cache_resource
 def load_text_model():
-    return pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
+    return pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
 
 # Function to process text with a language model
 def process_with_llm(prompt):
-    llm_model = load_text_model()
-    response = llm_model(prompt, max_length=500, do_sample=True, temperature=0.7)
-    return response[0]["generated_text"]
+    try:
+        llm_model = load_text_model()
+        response = llm_model(prompt, max_length=500, do_sample=True, temperature=0.7)
+        if response and len(response) > 0:
+            return response[0]["generated_text"]
+        else:
+            return "No explanation or completion could be generated. Please try again with a different input."
+    except Exception as e:
+        return f"Error generating explanation: {e}"
 
 # Streamlit App
 def main():
@@ -40,7 +46,7 @@ def main():
             # Extract text from image
             ocr_model = load_ocr_model()
             result = ocr_model(image)
-            
+
             if len(result) > 0:
                 extracted_text = result[0]["generated_text"]
                 st.write("### Extracted Text:")
@@ -49,8 +55,7 @@ def main():
                 # Process extracted text with LLM
                 st.write("### Explanation/Completion:")
                 explanation = process_with_llm(extracted_text)
-                if explanation:
-                    st.write(explanation)
+                st.write(explanation)
             else:
                 st.error("No text could be extracted. Please try another image.")
         except Exception as e: