add app.py

app.py

@@ -19,16 +19,16 @@ def filename_from_text(text: str) -> str:
 def log_gpu_memory():
     print(subprocess.check_output('nvidia-smi').decode('utf-8'))
 
-log_gpu_memory()
+# log_gpu_memory()
 
 model = MinDalle(
     is_mega=True, 
     is_reusable=True,
-    device='cuda',
-    dtype=torch.float32
+    device='cpu',
+    # dtype=torch.float32
 )
 
-log_gpu_memory()
+# log_gpu_memory()
 
 def run_model(
     text: str,
@@ -147,13 +147,15 @@ with demo:
 
             gradio.Markdown(
                 """
-                ####
+                #### Parameter
                 - **Input Text**: For long prompts, only the first 64 text tokens will be used to generate the image.
                 - **Grid Size**: Size of the image grid. 3x3 takes about 15 seconds.
                 - **Seamless**: Tile images in image token space instead of pixel space.
                 - **Temperature**: High temperature increases the probability of sampling low scoring image tokens.
                 - **Top-k**: Each image token is sampled from the top-k scoring tokens.
                 - **Super Condition**: Higher values can result in better agreement with the text.
+
+                #### 
                 """
             )