Update app.py

app.py

@@ -18,6 +18,15 @@ st.write("Waiting for image upload and prompt input...")
 uploaded_file = st.file_uploader("Upload an image (JPG or PNG):", type=["jpg", "jpeg", "png"])
 prompt = st.text_input("Enter your prompt:", "A little girl is riding a bicycle at high speed. Focused, detailed, realistic.")
 
+# Cache migration step (add this before model initialization)
+st.write("Migrating the cache for model files...")
+try:
+    from transformers.utils import move_cache
+    move_cache()
+    st.write("Cache migration completed successfully.")
+except Exception as e:
+    st.error(f"Cache migration failed: {e}")
+
 if uploaded_file and prompt:
     try:
         # Debug: File and prompt received