Spaces:

gmbreemer
/

spaces_streamlit

Runtime error

gmbreemer commited on Oct 10, 2022

Commit

88b0bf6

1 Parent(s): d6fd321

Remove typing

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ WANDB_RUN = "207pa091"
 def pred_and_plot_image(
     model: torch.nn.Module,
     image_file: BytesIO,
-    class_names: list[str],
     transform,
 ):
     """Run inference and show the results."""

 def pred_and_plot_image(
     model: torch.nn.Module,
     image_file: BytesIO,
+    class_names,
     transform,
 ):
     """Run inference and show the results."""