Updating Side-By-Side Space

app.py

@@ -33,7 +33,7 @@ def load_models(progress=None):
 
     try:
         # Load base model
-        if progress:
+        if progress is not None:
             progress(0.2, desc="Loading base model... This may take a few minutes")
 
         print(f"Loading base model from {BASE_MODEL_PATH}")
@@ -44,7 +44,7 @@ def load_models(progress=None):
         )
 
         # Load novel model
-        if progress:
+        if progress is not None:
             progress(0.7, desc="Loading novel model... This may take a few minutes")
 
         print(f"Loading novel model from {NOVEL_MODEL_PATH}")
@@ -54,7 +54,7 @@ def load_models(progress=None):
             n_threads=4  # Number of CPU threads to use
         )
 
-        if progress:
+        if progress is not None:
             progress(1.0, desc="Models loaded successfully!")
 
         print("Models loaded successfully!")
@@ -62,7 +62,7 @@ def load_models(progress=None):
 
     except Exception as e:
         print(f"Error loading models: {str(e)}")
-        if progress:
+        if progress is not None:
             progress(1.0, desc=f"Error loading models: {str(e)}")
         return False
 

requirements.txt

@@ -5,5 +5,4 @@ torch==2.7.0
 accelerate==1.6.0
 sentencepiece==0.2.0
 protobuf==6.30.2
-llama-cpp-python==0.3.9
---extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu
+llama-cpp-python==0.3.9