Parcourir la source

Merge pull request #441 from syyxsxx/restful

fix delete model bug
Jason il y a 5 ans
Parent
commit
b7ab73616d
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      paddlex/restful/app.py

+ 2 - 2
paddlex/restful/app.py

@@ -870,11 +870,11 @@ def model():
     if request.method == 'DELETE':
         if data['type'] == 'pretrained':
             from .model import delete_pretrained_model
-            ret = delete_pretrained_model(data, workspace)
+            ret = delete_pretrained_model(data, SD.workspace)
             return ret
         if data['type'] == 'exported':
             from .model import delete_exported_model
-            ret = delete_exported_model(data, workspace)
+            ret = delete_exported_model(data, SD.workspace)
             return ret