Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ def generate_model_list():
|
|
| 33 |
if "OVModelFor" in task:
|
| 34 |
cls = getattr(import_module("test_modeling"), test)
|
| 35 |
else:
|
| 36 |
-
cls = getattr(import_module("
|
| 37 |
try:
|
| 38 |
print(cls.SUPPORTED_ARCHITECTURES)
|
| 39 |
d[task] = cls.SUPPORTED_ARCHITECTURES
|
|
|
|
| 33 |
if "OVModelFor" in task:
|
| 34 |
cls = getattr(import_module("test_modeling"), test)
|
| 35 |
else:
|
| 36 |
+
cls = getattr(import_module("test_diffusion"), test)
|
| 37 |
try:
|
| 38 |
print(cls.SUPPORTED_ARCHITECTURES)
|
| 39 |
d[task] = cls.SUPPORTED_ARCHITECTURES
|