Update app.py

app.py

@@ -74,13 +74,13 @@ st.title("Inpainting App")
 st.markdown("This app allows you to inpaint images using a pre-trained model.")
 
 @st.cache(allow_output_mutation=True)
-def load_model():
+def load_model(BERT):
 
     # Load the model here...
 
     return model
 
-    def inpaint(image, GPTNEOX):
+    def inpaint(image, BERT):
 
     # Inpaint the image here...