Update app.py

app.py

@@ -54,9 +54,13 @@ def infer(
     prompt = str(prompt) if prompt else ""
     negative_prompt = str(negative_prompt) if negative_prompt else ""
 
+    # Convert text inputs to Long tensors if necessary
+    text_input_ids = torch.tensor([ord(c) for c in prompt], dtype=torch.long, device=device)
+    neg_text_input_ids = torch.tensor([ord(c) for c in negative_prompt], dtype=torch.long, device=device)
+
     image = pipe(
-        prompt=prompt,
-        negative_prompt=negative_prompt,
+        prompt=text_input_ids,
+        negative_prompt=neg_text_input_ids,
         guidance_scale=torch.tensor(guidance_scale, dtype=torch.float32, device=device),
         num_inference_steps=num_inference_steps,
         width=width,