Update app.py

app.py

@@ -55,8 +55,10 @@ def infer(
     negative_prompt = str(negative_prompt) if negative_prompt else ""
 
     # Ensure text input IDs are of type LongTensor
-    if hasattr(pipe.text_encoder, 'input_ids'):
-        pipe.text_encoder.input_ids = pipe.text_encoder.input_ids.to(torch.long)
+    if isinstance(prompt, torch.Tensor):
+        prompt = prompt.to(torch.long)
+    if isinstance(negative_prompt, torch.Tensor):
+        negative_prompt = negative_prompt.to(torch.long)
 
     image = pipe(
         prompt=prompt,
@@ -170,4 +172,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()