Update app.py

app.py

@@ -9,7 +9,7 @@ import argparse
 from einops import rearrange
 from fire import Fire
 from PIL import ExifTags, Image
-import spaces
+#import spaces
 
 import torch
 import torch.nn.functional as F
@@ -20,7 +20,7 @@ from transformers import pipeline
 from flux.sampling import denoise, get_schedule, prepare, unpack
 from flux.util import (configs, embed_watermark, load_ae, load_clip, load_flow_model, load_t5)
 from huggingface_hub import login
-login(token=os.getenv('Token'))
+#login(token=os.getenv('Token'))
 
 import torch
 
@@ -61,7 +61,7 @@ is_schnell = False
 output_dir = 'result'
 add_sampling_metadata = True
 
-@spaces.GPU(duration=120)
+#@spaces.GPU(duration=120)
 @torch.inference_mode()
 def edit(init_image, source_prompt, target_prompt, editing_strategy, num_steps, inject_step, guidance, seed):
     global ae, t5, clip, model, name, is_schnell, output_dir, add_sampling_metadata
@@ -269,4 +269,4 @@ def create_demo(model_name: str, device: str = "cuda:0" if torch.cuda.is_availab
     return demo
 
 demo = create_demo("flux-dev", "cuda")
-demo.launch()
+demo.launch(share = True)