Clear cache on inference failure

sketch2diagram.py

@@ -50,7 +50,12 @@ if input_file is not None:
 # Run model inference
 if generate_command:
     with st.spinner("Generating TikZ code..."):
-        output = run_inference(input_file, model_name, args)[0]
+        try:
+            output = run_inference(input_file, model_name, args)[0]
+        except Exception as e:
+            st.error(f"Inference failed: {e}")
+            st.session_state.clear()
+    return
         pdf_file_path = compile_tikz_to_pdf(output)
         if output and pdf_file_path:
             st.subheader("Generated TikZ Code")