Update app.py

app.py

@@ -38,6 +38,7 @@ titok_tokenizer, titok_generator = load_model()
 def demo_infer(
                guidance_scale, randomize_temperature, num_sample_steps,
                class_label, seed):
+    device = "cuda"
     # device = "cuda" if torch.cuda.is_available() else "cpu"
     tokenizer = titok_tokenizer #.to(device)
     generator = titok_generator #.to(device)