Update main.py

main.py

@@ -13,10 +13,10 @@ else:
     is_gpu = True
 print(is_gpu)
 
-
 from IPython.display import clear_output
 
 
+
 def check_enviroment():
     try:
         import torch
@@ -164,7 +164,7 @@ clear_output()
 from IPython.display import display
 
 
-def generate_image(prompt="A beautiful and sexy girl"):
+def generate_image(prompt="A beautiful and sexy girl",secret_token="default_secret"):
     # Generate the image using the prompt
     generated_image = generate(
         prompt=prompt,
@@ -174,7 +174,7 @@ def generate_image(prompt="A beautiful and sexy girl"):
         height=1024,
         guidance_scale=0.0,
         num_inference_steps=4,
-        secret_token="default_secret",  # Replace with your secret token
+        secret_token=secret_token
     )
     # Display the image in the Jupyter Notebook
     display(generated_image)