fixed bug arguments

app.py

@@ -10,7 +10,7 @@ from huggingface_hub import snapshot_download
 snapshot_download(repo_id="Roomie/xavyy", cache_dir='./')
 
 pipeline = AutoPipelineForText2Image.from_pretrained(
-    'black-forest-labs/FLUX.1-schnell', torch_dtype=torch.bfloat16).to('cpu')
+    'black-forest-labs/FLUX.1-schnell', torch_dtype=torch.bfloat16).to('cuda')
 pipeline.load_lora_weights('Roomie/xavyy', weight_name='xavyy.safetensors')
 
 
@@ -20,12 +20,7 @@ MAX_IMAGE_SIZE = 1024
 # @spaces.GPU #[uncomment to use ZeroGPU]
 
 
-def infer(prompt,  seed, randomize_seed):
-
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-
-    generator = torch.Generator().manual_seed(seed)
+def infer(prompt):
 
     image = pipeline(
         prompt=prompt,
@@ -37,7 +32,7 @@ def infer(prompt,  seed, randomize_seed):
         # generator=generator
     ).images[0]
 
-    return image, seed
+    return image
 
 
 examples = [