Upload app.py

app.py

@@ -177,7 +177,7 @@ def query_summarization(text):
     response = requests.request("POST", API_URL, headers=headers, json=payload).json()
     return response[0].get('summary_text')
 
-def generate_poster(movie_data, _api):
+def generate_poster(movie_data):
     """
     Function for recommending movies
     -param movie_data: metadata of movie selected by user
@@ -195,7 +195,7 @@ def generate_poster(movie_data, _api):
     # Load text-to-image model
     if not IS_MODEL_LOADED:
         with st.spinner("Loading Text to Image model..."):
-            pipeline = load_model(api)
+            pipeline = load_model()
     
     # Get image based on synopsis
     poster_image = pipeline(synopsis_sum).images[0]
@@ -216,10 +216,6 @@ if __name__ == "__main__":
     # Initialize image variable
     image = None
 
-    ## Create dataset
-    data = load_dataset()
-
-
     ## --- Page config ------------ ##
     # Set page title
     st.title("""
@@ -228,12 +224,12 @@ if __name__ == "__main__":
     Just select the title of a movie to generate an alternative poster.
      """)
 
-    ## Set page footer
+    # Set page footer
     footer()
     ## ---------------------------- ##
 
     ## --- Connect to kaggle API --- #
-
+    # Save key to json file
     if not os.path.exists(PATH_JSON):
         api_token = {"username":st.secrets['username'],"key":st.secrets['key']}
         with open(PATH_JSON, 'w') as file:
@@ -245,6 +241,8 @@ if __name__ == "__main__":
     api.authenticate()
     ## ---------------------------- ##
 
+    ## Create dataset
+    data = load_dataset(api)
 
     st.text("")
     st.text("")
@@ -262,7 +260,7 @@ if __name__ == "__main__":
 
 
     if is_clicked:
-        image = generate_poster(data[data.title==session.selected_movie], api)
+        image = generate_poster(data[data.title==session.selected_movie])
 
     st.text("")
     st.text("")