Upload streamlit_app.py

src/streamlit_app.py

@@ -110,24 +110,11 @@ div[data-testid="stFileUploader"] * {
 
 /* ═══════════════════════════════════════════════════════════════ */
 
-/* Stats & DYNAMIC Output Box */
+/* Stats & Output Box */
 .stat-card { background: #000; padding: 15px; border-radius: 4px; text-align: center; border: 1px solid rgba(143, 245, 255, 0.1); margin-bottom: 10px; }
 .stat-val { color: #8ff5ff; font-size: 24px; font-weight: 700; font-family: 'Space Grotesk'; }
 .stat-lbl { font-size: 9px; color: #46484d; text-transform: uppercase; letter-spacing: 2px; }
-
-.output-box { 
-    border-left: 3px solid #8ff5ff; 
-    background: #171a1f; 
-    padding: 25px; 
-    font-family: 'Space Grotesk'; 
-    font-size: 18px; 
-    line-height: 1.8; 
-    height: calc(100vh - 320px); /* Dynamically scales to viewport */
-    min-height: 400px; /* Safe fallback */
-    overflow-y: auto; 
-    white-space: pre-wrap; 
-    border-radius: 0 4px 4px 0;
-}
+.output-box { border-left: 3px solid #8ff5ff; background: #171a1f; padding: 25px; font-family: 'Space Grotesk'; font-size: 18px; line-height: 1.8; height: 400px; overflow-y: auto; white-space: pre-wrap; border-radius: 0 4px 4px 0;}
 
 .stButton>button { background-color: rgba(143, 245, 255, 0.05) !important; border: 1px solid #8ff5ff !important; color: #8ff5ff !important; width: 100%; padding: 12px; }
 .stButton>button:hover { background-color: #8ff5ff !important; color: #000 !important; }
@@ -149,9 +136,12 @@ def load_vision_engine():
 @st.cache_resource(show_spinner=False)
 def load_trocr_model(model_path):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    
+    # Hugging Face natively downloads the processor via the repo ID
     proc = TrOCRProcessor.from_pretrained(model_path)
     
     if os.path.exists(model_path):
+        # Local Loading Logic (Kept as a fallback just in case you run it locally again)
         config = VisionEncoderDecoderConfig.from_pretrained(model_path)
         model = VisionEncoderDecoderModel(config)
         safe_path = os.path.join(model_path, "model.safetensors")
@@ -163,6 +153,7 @@ def load_trocr_model(model_path):
         else:
             model.load_state_dict(torch.load(bin_path, map_location="cpu", weights_only=True), strict=False)
     else:
+        # Cloud Loading Logic: This natively pulls your model from the Hugging Face Hub
         model = VisionEncoderDecoderModel.from_pretrained(model_path)
         
     # Push standard registered parameters/buffers to device
@@ -240,9 +231,13 @@ def main():
     run_scan_trigger = False
     
     with c_left:
+        # ─── THE CLOUD REPOSITORY MAP ───
         model_choice = st.selectbox("SELECT MODEL", ["V13 Specialist", "Microsoft Large"])
         st.markdown("<div style='height: 15px;'></div>", unsafe_allow_html=True)
-        m_map = {"V13 Specialist": "./final_handwriting_model_v13", "Microsoft Large": "microsoft/trocr-large-handwritten"}
+        m_map = {
+            "V13 Specialist": "Hypernova823/ReadAI", 
+            "Microsoft Large": "microsoft/trocr-large-handwritten"
+        }
 
         if st.session_state.image_data is None:
             st.markdown("""