harrydawitch commited on
Commit
b4cc453
·
verified ·
1 Parent(s): 810485e

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +46 -0
app.py ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gc
3
+
4
+ import streamlit as st
5
+ from streamlit_option_menu import option_menu
6
+
7
+ from app.exterior import app as exterior_app
8
+ from app.homepage import app as homepage_app
9
+ from app.interior import app as interior_app
10
+ from backend.inference import ArchIntelligent
11
+
12
+
13
+
14
+ st.set_page_config(page_title="ArchIntelligent", page_icon="🏠", initial_sidebar_state='expanded')
15
+
16
+
17
+
18
+
19
+ @st.cache_resource
20
+ def load_model():
21
+ with st.spinner("🔄 Loading base model...", show_time= True):
22
+ gc.enable()
23
+ gc.collect()
24
+ return ArchIntelligent()
25
+
26
+ pipe = load_model()
27
+
28
+
29
+ # with st.sidebar:
30
+ selected = option_menu(
31
+ menu_title="Main Menu",
32
+ options=["Home", "Exterior Design", "Interior Design"],
33
+ icons=["Psychology", "Home", "Bed"],
34
+ orientation= 'horizontal',
35
+ menu_icon="list",
36
+ default_index=0,
37
+
38
+ )
39
+
40
+
41
+ if selected == "Home":
42
+ homepage_app()
43
+ elif selected == "Exterior Design":
44
+ exterior_app(pipe)
45
+ elif selected == "Interior Design":
46
+ interior_app()