readme

README.md

@@ -43,7 +43,7 @@ At AiArtLab, we strive to create a free, compact and fast model that can be trai
 
 ### Text 2 image
 
-```
+```python
 #!pip install -U torch torchvision
 #!pip install -U diffusers accelerate transformers
 
@@ -106,7 +106,7 @@ Development Note: We have not provided direct comparisons with other popular ups
 
 ### upscale code example
 
-```
+```python
 import torch
 import numpy as np