Merge pull request #83 from berylliumsec/enable_loading_indicator

Add busy indicator
berylliumsec · Feb 5, 2025 · 07213db · 07213db
2 parents 0016688 + cf71f00
commit 07213db
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -122,7 +122,7 @@ jobs:
           prerelease: false
           body: |
             Release Notes:
-              This release ensures that models always return output even if the marker that differentiates between prompts and text generation are not found
+              This release adds an indicator to show that the model is busy upon startup
       - name: Upload Release Assets
         if: github.event.pull_request.merged == true
         run: |

diff --git a/src/nebula/terminal_emulator.py b/src/nebula/terminal_emulator.py
@@ -1487,12 +1487,14 @@ def startModelCreation(self):
 
     def onModelCreated(self, model):
         if model:
+            self.threads_status.emit("completed")
             self.model = model  # Now you have your model ready to be used
             self.model_created.emit(True)
 
     def create_model(self, status):
 
         if status is True and self.model is None:
+            self.threads_status.emit("in_progress")
             logger.debug("loading model")
 
             self.startModelCreation()