helenai commited on
Commit
650b879
1 Parent(s): 47b430f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("test_stable_diffusion"), test)
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