Update app.py

app.py

@@ -1,3 +1,4 @@
+# Install required packages
 !pip install sentencepiece
 !pip install git+https://github.com/huggingface/transformers.git@cae78c46
 !pip install diffusers
@@ -8,19 +9,18 @@
 !pip install gradio==4.12.0
 !pip install gradio_client==0.8.0
 !pip install -i https://download.pytorch.org/whl/cu118 torch==2.0 torchvision==0.15 torchaudio==2.0
-#conversation.py:
+
+# conversation.py
 import dataclasses
 from enum import auto, Enum
 from typing import List, Tuple
 
-
 class SeparatorStyle(Enum):
     """Different separator style."""
     SINGLE = auto()
     TWO = auto()
     MPT = auto()
 
-
 @dataclasses.dataclass
 class Conversation:
     """A class that keeps all conversation history."""
@@ -376,7 +376,8 @@ conv_templates = {
 
 if __name__ == "__main__":
     print(default_conversation.get_prompt())
-#mgie_llava.py:
+
+# mgie_llava.py
 from typing import List, Optional, Tuple, Union
 
 import torch
@@ -397,11 +398,9 @@ DEFAULT_IMAGE_PATCH_TOKEN = "<im_patch>"
 DEFAULT_IM_START_TOKEN = "<im_start>"
 DEFAULT_IM_END_TOKEN = "<im_end>"
 
-
 class LlavaConfig(LlamaConfig):
     model_type = "llava"
 
-
 class LlavaLlamaModel(LlamaModel):
     config_class = LlavaConfig
 
@@ -778,7 +777,8 @@ class LlavaLlamaForCausalLM(LlamaForCausalLM):
 
 AutoConfig.register("llava", LlavaConfig)
 AutoModelForCausalLM.register(LlavaConfig, LlavaLlamaForCausalLM)
-#main.py:
+
+# main.py
 from google.colab import drive
 drive.mount('/content/drive')
 
@@ -793,9 +793,6 @@ import huggingface_hub
 
 CKPT_DIR = '/content/drive/My Drive/_ckpt'
 
-
-
-
 def crop_resize(f, sz=512):
     w, h = f.size
     if w > h:
@@ -908,4 +905,4 @@ with gr.Blocks() as app:
         btn_sub = gr.Button('Submit')
     btn_sub.click(fn=go_mgie, inputs=[inp, txt, seed, cfg_txt, cfg_img], outputs=[res, out])
 
-app.launch()
+app.launch()