Update app.py
Browse files
app.py
CHANGED
|
@@ -18,6 +18,12 @@ if gr.__version__ != "3.28.3": #doesn't work...
|
|
| 18 |
os.system("pip uninstall -y gradio")
|
| 19 |
os.system("pip install gradio==3.28.3")
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
description = """
|
| 22 |
Our project is to use diffusion model to change the texture of our robotic arm simulation.
|
| 23 |
To do so, we first get our simulated images. After, we process these images to get Canny Edge maps. Finally, we can get brand new images by using ControlNet.
|
|
@@ -188,7 +194,7 @@ def preprocess_canny(image, resolution=128):
|
|
| 188 |
def create_demo(process, max_images=12, default_num_images=4):
|
| 189 |
with gr.Blocks() as demo:
|
| 190 |
with gr.Row():
|
| 191 |
-
gr.Markdown(
|
| 192 |
with gr.Row():
|
| 193 |
with gr.Column():
|
| 194 |
input_image = gr.Image(source='upload', type='numpy')
|
|
|
|
| 18 |
os.system("pip uninstall -y gradio")
|
| 19 |
os.system("pip install gradio==3.28.3")
|
| 20 |
|
| 21 |
+
title_description = """
|
| 22 |
+
# SynDRoM
|
| 23 |
+
## Synthetic Data augmentation for Robotic Manipulation
|
| 24 |
+
|
| 25 |
+
"""
|
| 26 |
+
|
| 27 |
description = """
|
| 28 |
Our project is to use diffusion model to change the texture of our robotic arm simulation.
|
| 29 |
To do so, we first get our simulated images. After, we process these images to get Canny Edge maps. Finally, we can get brand new images by using ControlNet.
|
|
|
|
| 194 |
def create_demo(process, max_images=12, default_num_images=4):
|
| 195 |
with gr.Blocks() as demo:
|
| 196 |
with gr.Row():
|
| 197 |
+
gr.Markdown(title_description)
|
| 198 |
with gr.Row():
|
| 199 |
with gr.Column():
|
| 200 |
input_image = gr.Image(source='upload', type='numpy')
|