Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 import random
 import logging
 import numpy as np
+import spaces
 from typing import Any, Dict, List, Optional, Union
 
 import torch
@@ -28,16 +29,6 @@ from huggingface_hub import (
 
 from diffusers.utils import load_image
 
-import spaces
-
-#---if workspace = local or colab---
-
-# Authenticate with Hugging Face
-# from huggingface_hub import login
-
-# Log in to Hugging Face using the provided token
-# hf_token = 'hf-token-authentication'
-# login(hf_token)
 
 def calculate_shift(
     image_seq_len,
@@ -1286,7 +1277,7 @@ css = '''
 .progress-bar {height: 100%;background-color: #4f46e5;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out}
 '''
 
-with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 60)) as app:
+with gr.Blocks(theme="bethecloud/storj_theme", css=css, delete_cache=(60, 60)) as app:
     title = gr.HTML(
         """<h1>FLUX LoRA DLC2🔥</h1>""",
         elem_id="title",