fire2

app.py

@@ -9,17 +9,17 @@ def load_model():
     return loaded_pipeline
 
 # 예측 함수 정의
-def predict_cost(구조, 지하층수, 지상층수, 주차수,  연면적, 대지면적, 건축면적, 조경면적, 발주년, 발주월, 공사기간, 현장위치, 건축법상용도):
+def predict_cost(구조, 지하층수, 지상층수, 주차대수,  연면적, 대지면적, 건축면적, 조경면적, 발주년도, 발주월, 공사기간, 현장위치, 건축법상용도):
     new_project = {
         '구조': [구조],
         '지하층수': [지하층수],
         '지상층수': [지상층수],
-        '주차수': [주차수],
+        '주차대수': [주차대수],
         '연면적': [연면적],
         '대지면적': [대지면적],
         '건축면적': [건축면적],
         '조경면적': [조경면적],
-        '발주년': [발주년],
+        '발주년도': [발주년도],
         '발주월': [발주월],
         '공사기간': [공사기간],
         '현장위치': [현장위치],
@@ -31,7 +31,7 @@ def predict_cost(구조, 지하층수, 지상층수, 주차수,  연면적, 대
     return [f'{cost:,.0f}원' for cost in pred[0]]
 
 with gr.Blocks() as demo:
-    gr.Markdown("HG 공사비 예측 모델")
+    gr.Markdown("## HG 공사비 예측 모델")
     with gr.Row():
         구조 = gr.Dropdown(label="구조", choices=['철근콘크리트조', '철골조', '철골철근콘크리트조', '기타'], value='철근콘크리트조')  # '...'부분에 다른 구조 타입 추가
     with gr.Row():
@@ -58,7 +58,7 @@ with gr.Blocks() as demo:
        '업무시설 ', '연구시설', '운동시설', '운수시설', '운수시설 ', '의료시설', '자동차관련시설',
        '자원순환관련시설', '전시시설', '주차장', '창고시설', '체육시설', '판매시설'], value='업무시설')  # '...'부분에 다른 건축법상용도 추가
     
-    btn = gr.Button("예측")
+    btn = gr.Button("공사비 예측")
     
     btn.click(
         predict_cost,
@@ -70,6 +70,7 @@ with gr.Blocks() as demo:
             gr.Textbox(label="기계공사비"),
             gr.Textbox(label="전기공사비"),
             gr.Textbox(label="통신공사비"),
+            gr.Textbox(label="소방공사비"),
             gr.Textbox(label="토목공사비"),
             gr.Textbox(label="조경공사비")
         ]

model.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5b78ecb24082f8df462e159beb45a3e225f82722de4d346813f29cd572170b97
-size 10986993
+oid sha256:c16956282fcebe88c53f3a5aa2a883b6cf49fffc20f2745dc6fb188944e04709
+size 10986995