flagging dir uses function name

brstar96 · Jul 28, 2020 · 05099e7 · 05099e7
1 parent 593af4e
commit 05099e7
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/gradio/interface.py b/gradio/interface.py
@@ -125,17 +125,14 @@ def get_output_instance(iface):
 
         if self.allow_flagging:
             if self.title is not None:
-                dir_name = "-".join(self.title.split(" ")) + "-1"
+                dir_name = "_".join(self.title.split(" ")) + "_1"
             else:
-                dir_name = "-".join([inp.label for inp in
-                                     self.input_interfaces]) + "-to-" \
-                           + "-".join([out.label for out in
-                                       self.output_interfaces])\
-                           + "-1"
+                dir_name = "_".join([fn.__name__ for fn in self.predict]) + \
+                           "_1"
             i = 1
             while os.path.exists(FLAGGING_DIRECTORY + dir_name):
                 i += 1
-                dir_name = dir_name[:-2] + "-" + str(i)
+                dir_name = dir_name[:-2] + "_" + str(i)
             self.flagging_dir = FLAGGING_DIRECTORY + dir_name
 
         try: