Merge pull request #69 from JSchmie/develop

Minor bug fixes

pyproject.toml

@@ -50,7 +50,7 @@ strict = true
 format-jinja = """
     {%- if distance == 0 -%}
         {{ serialize_pep440(base) }}
-    {%- elif branch == 'develop' -%}
+    {%- elif branch == 'develop' or branch == 'main' -%}
         {{ serialize_pep440(bump_version(base), dev = distance) }}
     {%- else -%}
         {{ serialize_pep440(bump_version(base), dev=distance, metadata=[commit]) }}

scraibe_webui/utils/background.py

@@ -5,7 +5,8 @@
 from os.path import join, split
 
 from threading import Thread, BoundedSemaphore, active_count
-from torch import set_num_threads
+from scraibe.misc import set_threads
+
 from scraibe_webui.global_var import MAX_CONCURRENT_MODELS
 import scraibe_webui.global_var as gv
 from .mail import MailService
@@ -63,8 +64,8 @@ def parrallel_task(self,
 
         """ Background task that runs in a separate thread """
 
-        if self.threads_per_model != 0:
-            set_num_threads(self.threads_per_model)
+        if self.threads_per_model  is not None:
+            set_threads(yaml_threads = self.threads_per_model) 
 
         # List to store temporary files
         temp_files = []
@@ -156,8 +157,8 @@ def parrallel_task(self,
                             temp_file.write(result)
 
                         temp_files.append(temp_file_path_json)
-            
-            MailService.from_config(self.mail_service_params).send_transcript(receiver_email=reciever, transcript_path = temp_files, **success_format_option)
+
+            MailService.from_config(self.mail_service_params).send_transcript(receiver_email=reciever, transcript_paths = temp_files, **success_format_option)
 
         except Exception as exeption:
 

scraibe_webui/utils/mail.py

@@ -217,7 +217,6 @@ def send_transcript(self, receiver_email: str, transcript_paths: Union[str, list
             transcript_paths = []
         elif isinstance(transcript_paths, str):
             transcript_paths = [transcript_paths]
-
         _message = (self.success_template or "Your transcript is ready.").format(
             css_path=self.css_template_path, **format_options
         )