Upload app.py

app.py

@@ -133,7 +133,7 @@ def authenticate_kaggle():
 
 
 @st.experimental_memo(persist=True, show_spinner=False, suppress_st_warning=True)
-def load_dataset():
+def load_dataset(_api):
     """
     Load Dataset from Kaggle
     -return: dataframe containing dataset
@@ -156,7 +156,7 @@ def load_dataset():
     return data
 
 @st.cache(persist=True, show_spinner=False, allow_output_mutation=True, suppress_st_warning=True)
-def load_model(_api):
+def load_model():
     model = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
     IS_MODEL_LOADED = True
     return model