Spaces:

xiongjie
/

face-expression-ja-example

Runtime error

xiongjie commited on Nov 7, 2021

Commit

a479071

1 Parent(s): b541280

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ model_args = ClassificationArgs()
 model_original = ClassificationModel(
     'auto',
     'xiongjie/face-expression-ja',
-    use_cuda=True,
     args=model_args,
     num_labels=len(CLASS_NAMES)
 )

 model_original = ClassificationModel(
     'auto',
     'xiongjie/face-expression-ja',
+    use_cuda=False,
     args=model_args,
     num_labels=len(CLASS_NAMES)
 )