fix

.idea/workspace.xml

@@ -7,6 +7,7 @@
     <list default="true" id="d268fe54-cb43-4451-8f43-ce4d20d5699c" name="Changes" comment="">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/Kaggle_API.py" beforeDir="false" afterPath="$PROJECT_DIR$/Kaggle_API.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/app.py" beforeDir="false" afterPath="$PROJECT_DIR$/app.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -61,7 +62,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1713919907805</updated>
-      <workItem from="1713919910236" duration="599000" />
+      <workItem from="1713919910236" duration="767000" />
     </task>
     <servers />
   </component>

Kaggle_API.py

@@ -5,11 +5,11 @@ from Helper_functions import *
 
 
 class API_Connection:
-    def __init__(self, gd_connection: GoogleDrive_API, kaggle_username: str = "", kaggle_key: str = ""):
+    def __init__(self, gd_connection, kaggle_username: str = "", kaggle_key: str = ""):
         os.environ["KAGGLE_USERNAME"] = kaggle_username
         os.environ["KAGGLE_KEY"] = kaggle_key
 
-        self.GoogleDrive_connection: GoogleDrive_API = gd_connection
+        self.GoogleDrive_connection = gd_connection
 
         self.NOTEBOOK_ID = "amirmorris/pix2pix-model"
         self.DATASET_NAME = "dataset"

app.py

@@ -53,7 +53,7 @@ def generate_button_clicked(*args):
         raise gr.Error("Missing Input: edit_instruction")
 
     GoogleDrive_connection = ""
-    # GoogleDrive_connection = GoogleDrive_API("service_account.json")
+    GoogleDrive_connection = GoogleDrive_API("service_account.json")
     api_connection = API_Connection(GoogleDrive_connection, kaggle_username, kaggle_key)
 
     create_folder("local_dataset")