Spaces:
Runtime error
Runtime error
Update app_dialogue.py
Browse files- app_dialogue.py +3 -1
app_dialogue.py
CHANGED
|
@@ -2,6 +2,7 @@ import copy
|
|
| 2 |
import os
|
| 3 |
import spaces
|
| 4 |
import subprocess
|
|
|
|
| 5 |
import torch
|
| 6 |
|
| 7 |
from threading import Thread
|
|
@@ -316,7 +317,8 @@ def model_inference(
|
|
| 316 |
except Exception as e:
|
| 317 |
print("error")
|
| 318 |
gr.Error(e)
|
| 319 |
-
print("
|
|
|
|
| 320 |
|
| 321 |
|
| 322 |
# Hyper-parameters for generation
|
|
|
|
| 2 |
import os
|
| 3 |
import spaces
|
| 4 |
import subprocess
|
| 5 |
+
import time
|
| 6 |
import torch
|
| 7 |
|
| 8 |
from threading import Thread
|
|
|
|
| 317 |
except Exception as e:
|
| 318 |
print("error")
|
| 319 |
gr.Error(e)
|
| 320 |
+
print(f"Success! Generated the following sequence: `{acc_text}`")
|
| 321 |
+
time.sleep(0.5)
|
| 322 |
|
| 323 |
|
| 324 |
# Hyper-parameters for generation
|