Update app.py

app.py

@@ -39,14 +39,12 @@ MAX_IMAGE_SIZE = 2048
 @spaces.GPU
 def infer(seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
     prompt = "<|bos|><rating>rating:sfw, rating:general</rating><copyright>original</copyright><character></character><general><|long|>1girl<|input_end|>"
-    inputs = tokenizer(prompt, return_tensors="pt").input_ids
+    inputs = tokenizer(prompt, return_tensors="pt").input_ids.to(device)
     
     with torch.no_grad():
       outputs = model.generate(inputs, generation_config=model.generation_config)
     
     prompt=tokenizer.decode(outputs[0], skip_special_tokens=True).split("rating:general,")[1]
-    # rating:sfw, rating:general, 1girl, ahoge, braid, closed eyes, collared dress, dress, flower, full body, hair flower, hair ornament, long hair, night, night sky, outdoors, parted lips, pink flower, pink hair, short sleeves, sky, solo, straight hair, sunflower, very long hair, white flower
-
     negative_prompt="unaestheticXLv31"
     
     if randomize_seed: