debug

app.py

@@ -17,6 +17,15 @@ if not torch.cuda.is_available():
 # model = Model(base_model_id=DEFAULT_MODEL_ID, task_name="Canny")
 model = Model(base_model_id=DEFAULT_MODEL_ID, task_name="texnet")
 
+import sys
+pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
+version_str="".join([
+    f"py3{sys.version_info.minor}_cu",
+    torch.version.cuda.replace(".",""),
+    f"_pyt{pyt_version_str}"
+])
+print(f"Using version: {version_str}")
+
 with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
     gr.DuplicateButton(