Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,6 @@ def load_model():
|
|
| 35 |
)
|
| 36 |
return text_to_image
|
| 37 |
|
| 38 |
-
text_to_image = load_model()
|
| 39 |
|
| 40 |
code = r'''
|
| 41 |
import paramiko
|
|
@@ -73,6 +72,7 @@ def infer_30(
|
|
| 73 |
num_inference_steps,
|
| 74 |
progress=gr.Progress(track_tqdm=True),
|
| 75 |
):
|
|
|
|
| 76 |
os.environ['JAX_PLATFORMS'] = 'gpu'
|
| 77 |
os.environ['KERAS_BACKEND'] = 'jax'
|
| 78 |
seed = random.randint(0, MAX_SEED)
|
|
@@ -104,6 +104,7 @@ def infer_60(
|
|
| 104 |
num_inference_steps,
|
| 105 |
progress=gr.Progress(track_tqdm=True),
|
| 106 |
):
|
|
|
|
| 107 |
os.environ['JAX_PLATFORMS'] = 'gpu'
|
| 108 |
os.environ['KERAS_BACKEND'] = 'jax'
|
| 109 |
seed = random.randint(0, MAX_SEED)
|
|
@@ -135,6 +136,7 @@ def infer_90(
|
|
| 135 |
num_inference_steps,
|
| 136 |
progress=gr.Progress(track_tqdm=True),
|
| 137 |
):
|
|
|
|
| 138 |
os.environ['JAX_PLATFORMS'] = 'gpu'
|
| 139 |
os.environ['KERAS_BACKEND'] = 'jax'
|
| 140 |
seed = random.randint(0, MAX_SEED)
|
|
|
|
| 35 |
)
|
| 36 |
return text_to_image
|
| 37 |
|
|
|
|
| 38 |
|
| 39 |
code = r'''
|
| 40 |
import paramiko
|
|
|
|
| 72 |
num_inference_steps,
|
| 73 |
progress=gr.Progress(track_tqdm=True),
|
| 74 |
):
|
| 75 |
+
text_to_image = load_model()
|
| 76 |
os.environ['JAX_PLATFORMS'] = 'gpu'
|
| 77 |
os.environ['KERAS_BACKEND'] = 'jax'
|
| 78 |
seed = random.randint(0, MAX_SEED)
|
|
|
|
| 104 |
num_inference_steps,
|
| 105 |
progress=gr.Progress(track_tqdm=True),
|
| 106 |
):
|
| 107 |
+
text_to_image = load_model()
|
| 108 |
os.environ['JAX_PLATFORMS'] = 'gpu'
|
| 109 |
os.environ['KERAS_BACKEND'] = 'jax'
|
| 110 |
seed = random.randint(0, MAX_SEED)
|
|
|
|
| 136 |
num_inference_steps,
|
| 137 |
progress=gr.Progress(track_tqdm=True),
|
| 138 |
):
|
| 139 |
+
text_to_image = load_model()
|
| 140 |
os.environ['JAX_PLATFORMS'] = 'gpu'
|
| 141 |
os.environ['KERAS_BACKEND'] = 'jax'
|
| 142 |
seed = random.randint(0, MAX_SEED)
|