diff --git a/lama_cleaner/app/src/App.tsx b/lama_cleaner/app/src/App.tsx
index 198f689..445f2fb 100644
--- a/lama_cleaner/app/src/App.tsx
+++ b/lama_cleaner/app/src/App.tsx
@@ -123,7 +123,7 @@ function App() {
return (
+ const renderFileSelect = () => {
+ return (
+
+ {
+ setFile(f)
+ }}
+ />
+
+ )
+ }
+
+ const renderCanvas = () => {
+ return (
{
if (r) {
@@ -1024,6 +1035,17 @@ export default function Editor(props: EditorProps) {
)}
+ )
+ }
+
+ return (
+
+ {file === undefined ? renderFileSelect() : renderCanvas()}
{showBrush && !isInpainting && !isPanning && (
@@ -1037,7 +1059,7 @@ export default function Editor(props: EditorProps) {
)}
- {isSD ? (
+ {isSD || file === undefined ? (
<>>
) : (
{
const renderHeader = () => {
return (