Update app.py

app.py

@@ -408,13 +408,14 @@ def main():
         tester.define_model()
         tester.load_data(img_path)
         tester.load_model(args.pretrained_path)
-        tab1, tab2 = st.tabs(["Texture Synthesis", "Texture Editing"])
-        with tab1:
-            st.header("Texture Synthesis")
-            tester.display_synthesis()
-        with tab2:
+        app_idx = st.selectbox('Please select between texture synthesis or editing',
+                              ["Texture Synthesis", "Texture Editing"])
+        if app_idx == 'Texture Editing':
             st.header("Texture Editing")
             tester.display_editing()
+        else:
+            st.header("Texture Synthesis")
+            tester.display_synthesis()
 
 if __name__ == '__main__':
     os.system("pip install torch-geometric==1.7.2")