Update sketch2diagram.py

sketch2diagram.py

@@ -55,7 +55,7 @@ if generate_command:
         except Exception as e:
             st.error(f"Inference failed: {e}")
             st.session_state.clear()
-    return
+            return
         pdf_file_path = compile_tikz_to_pdf(output)
         if output and pdf_file_path:
             st.subheader("Generated TikZ Code")