Update app.py

app.py

@@ -56,9 +56,9 @@ def infer(
 
     # Ensure text input IDs are of type LongTensor
     if isinstance(prompt, torch.Tensor):
-        prompt = prompt.to(torch.long)
+        prompt = prompt.to(torch.long).tolist()
     if isinstance(negative_prompt, torch.Tensor):
-        negative_prompt = negative_prompt.to(torch.long)
+        negative_prompt = negative_prompt.to(torch.long).tolist()
 
     image = pipe(
         prompt=prompt,