Fix loading issues in the App

app.py

@@ -1,15 +1,17 @@
-import streamlit as st
-import pandas as pd
-import numpy as np
 import os
+
 import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
+import streamlit as st
 from transformers import CLIPProcessor
+
 from medclip.modeling_hybrid_clip import FlaxHybridCLIP
 
 
 @st.cache(allow_output_mutation=True)
 def load_model():
-    model = FlaxHybridCLIP.from_pretrained("flax-community/medclip-roco")
+    model = FlaxHybridCLIP.from_pretrained("flax-community/medclip-roco", _do_init=False)
     processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
     return model, processor