Update app.py

app.py

@@ -41,7 +41,7 @@ def generate_image(text,folder, _ch3, images):
     text = text.replace("\n", "").replace("\t", "")
     text_encode =  [j.encode() for j in text.split(' ')]
     eval_text_encode, eval_len_text = model.netconverter.encode(text_encode)
-    eval_text_encode = eval_text_encode.to('cuda').repeat(batch_size, 1, 1)
+    eval_text_encode = eval_text_encode.to(DEVICE).repeat(batch_size, 1, 1)
 
     input_styles, page_val = model._generate_page(style_inputs.to(DEVICE).clone(), width_length, eval_text_encode, eval_len_text, no_concat = True)
     page_val = crop_(page_val[0]*255)