Update app.py

app.py

@@ -40,10 +40,10 @@ def get_lora_sd_pipeline(
 
 def process_prompt(prompt, tokenizer, text_encoder, max_length=77):
     tokens = tokenizer(prompt, truncation=False, return_tensors="pt")["input_ids"]
-    chunks = [tokens[:, i:i + max_length] for i in range(0, tokens.shape[1], max_length)]
+    part_s = [tokens[:, i:i + max_length] for i in range(0, tokens.shape[1], max_length)]
     
     with torch.no_grad():
-        embeds = [text_encoder(chunk.to(text_encoder.device))[0] for chunk in chunks]
+        embeds = [text_encoder(part.to(text_encoder.device))[0] for part in part_s]
     
     return torch.cat(embeds, dim=1)