clean up playground

Former-commit-id: a178953a6c

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "1.7.6"
+version = "1.7.7"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]

swarms/models/mistral.py

@@ -47,9 +47,12 @@ class Mistral:
         task: str
     ):
         """Run the model on a given task."""
-        
+
         try:
-            model_inputs = self.tokenizer([task], return_tensors="pt").to(self.device)
+            model_inputs = self.tokenizer(
+                [task], 
+                return_tensors="pt"
+            ).to(self.device)
             generated_ids = self.model.generate(
                 **model_inputs, 
                 max_length=self.max_length,