update code

.gitignore

@@ -0,0 +1 @@
+cache/*

src/app.py

@@ -31,7 +31,7 @@ if os.environ.get('SPACE_ID'):
 
 
 import streamlit as st
-from FineTune.model import ComputeScore
+from FineTune.model import ComputeStat
 import time
 
 # -----------------
@@ -95,7 +95,7 @@ def load_model(from_pretrained, base_model, cache_dir, device):
         cache_dir = cache_dir
     
     with st.spinner("🔄 Loading model... This may take a moment on first launch."):
-        model = ComputeScore.from_pretrained(
+        model = ComputeStat.from_pretrained(
             from_pretrained, 
             base_model, 
             base_model,