Update app.py

app.py

@@ -113,7 +113,6 @@ class SpatialAttnProcessor2_0(torch.nn.Module):
         temb=None):
         # un_cond_hidden_states, cond_hidden_states = hidden_states.chunk(2)
         # un_cond_hidden_states = self.__call2__(attn, un_cond_hidden_states,encoder_hidden_states,attention_mask,temb)
-        # 生成一个0到1之间的随机数
         global total_count,attn_count,cur_step,mask1024,mask4096
         global sa32, sa64
         global write
@@ -124,7 +123,6 @@ class SpatialAttnProcessor2_0(torch.nn.Module):
             self.id_bank[cur_step] = [hidden_states[:self.id_length], hidden_states[self.id_length:]]
         else:
             encoder_hidden_states = torch.cat((self.id_bank[cur_step][0].to(self.device),hidden_states[:1],self.id_bank[cur_step][1].to(self.device),hidden_states[1:]))
-        # 判断随机数是否大于0.5
         if cur_step <=1:
             hidden_states = self.__call2__(attn, hidden_states,None,attention_mask,temb)
         else:   # 256 1024 4096
@@ -377,12 +375,11 @@ css = '''
 
 #################################################
 title = r"""
-<h1 align="center">StoryDiffusion: Consistent Self-Attention for Long-Range Image and Video Generation</h1>
+<h1 align="center">Ai Comic Generator</h1>
 """
 
 description = r"""
-<b>Official 🤗 Gradio demo</b> for <a href='https://github.com/HVision-NKU/StoryDiffusion' target='_blank'><b>StoryDiffusion: Consistent Self-Attention for Long-Range Image and Video Generation</b></a>.<br>
-❗️❗️❗️[<b>Important</b>] Personalization steps:<br>
+<br>❗️❗️❗️[<b>Important</b>] Personalization steps:<br>
 1️⃣ Enter a Textual Description for Character, if you add the Ref-Image, making sure to <b>follow the class word</b> you want to customize with the <b>trigger word</b>: `img`, such as: `man img` or `woman img` or `girl img`.<br>
 2️⃣ Enter the prompt array, each line corrsponds to one generated image.<br>
 3️⃣ Choose your preferred style template.<br>
@@ -390,32 +387,14 @@ description = r"""
 """
 
 article = r"""
-If StoryDiffusion is helpful, please help to ⭐ the <a href='https://github.com/HVision-NKU/StoryDiffusion' target='_blank'>Github Repo</a>. Thanks! 
-[![GitHub Stars](https://img.shields.io/github/stars/HVision-NKU/StoryDiffusion?style=social)](https://github.com/HVision-NKU/StoryDiffusion)
----
-📝 **Citation**
-<br>
-If our work is useful for your research, please consider citing:
-```bibtex
-@article{Zhou2024storydiffusion,
-  title={StoryDiffusion: Consistent Self-Attention for Long-Range Image and Video Generation},
-  author={Zhou, Yupeng and Zhou, Daquan and Cheng, Ming-Ming and Feng, Jiashi and Hou, Qibin},
-  year={2024}
-}
-```
-📋 **License**
-<br>
-The Contents you create are under Apache-2.0 LICENSE. The Code are under Attribution-NonCommercial 4.0 International. 
-📧 **Contact**
-<br>
-If you have any questions, please feel free to reach me out at <b>[email protected]</b>.
+<br>If you have any questions, please feel free to reach me out at <b>huzefa.ahmed.web@gmail.com</b>.
 """
 version = r"""
-<h3 align="center">StoryDiffusion Version 0.01 (test version)</h3>
+<h3 align="center">Ai Comic Generator</h3>
 <h5 >1. Support image ref image. (Cartoon Ref image is not support now)</h5>
 <h5 >2. Support Typesetting Style and Captioning.(By default, the prompt is used as the caption for each image. If you need to change the caption, add a # at the end of each line. Only the part after the # will be added as a caption to the image.)</h5>
 <h5 >3. [NC]symbol (The [NC] symbol is used as a flag to indicate that no characters should be present in the generated scene images. If you want do that, prepend the "[NC]" at the beginning of the line. For example, to generate a scene of falling leaves without any character, write: "[NC] The leaves are falling."),Currently, support is only using Textual Description</h5>
-<h5>Tips: Not Ready Now! Just Test! It's better to use prompts to assist in controlling the character's attire. Depending on the limited code integration time, there might be some undiscovered bugs. If you find that a particular generation result is significantly poor, please email me (ypzhousdu@gmail.com)  Thank you very much.</h4>
+<h5>Tips: Not Ready Now! Just Test! It's better to use prompts to assist in controlling the character's attire. Depending on the limited code integration time, there might be some undiscovered bugs. If you find that a particular generation result is significantly poor, please email me (huzefa.ahmed.web@gmail.com)  Thank you very much.</h4>
 """
 #################################################
 global attn_count, total_count, id_length, total_length,cur_step, cur_model_type