Merge remote-tracking branch 'refs/remotes/origin/main'

README.md

@@ -1,5 +1,5 @@
 ---
-title: STC Demo V2
+title: STC Demo V2 DEV
 emoji: 🐢
 colorFrom: green
 colorTo: pink

app.py

@@ -50,13 +50,17 @@ class HfModelWrapper:
         streamer = TextIteratorStreamer(self.tokenizer, skip_prompt=True, skip_special_tokens=True)
 
         gen_kwargs = {
-            "inputs": input_ids,
+            "input_ids": input_ids,
             "streamer": streamer,
-            "pad_token_id": self.tokenizer.eos_token_id,
-            "max_length": 32768,
-            "temperature": 0.1,
-            "top_p": 0.8,
+            "max_new_tokens": 2048,  
+            "min_length": 10,
+            "temperature": 0.7,
+            "top_p": 0.9,
+            "top_k": 50,
             "repetition_penalty": 1.1,
+            "do_sample": True,
+            "eos_token_id": self.tokenizer.eos_token_id,
+            "pad_token_id": self.tokenizer.pad_token_id, 
         }
 
         thread = Thread(target=self.model.generate, kwargs=gen_kwargs)
@@ -76,6 +80,13 @@ class ArtemisiaTheme(Base):
             neutral_hue=colors.gray,
             **kwargs
         )
+
+<<<<<<< HEAD
+=======
+        # Set text colors to white
+        self.set(
+            body_text_color="#ffffff",
+        )
         
         # Background settings
         self.body_background_fill = "#4f008c"
@@ -95,6 +106,8 @@ class ArtemisiaTheme(Base):
         self.text_color = "#ffffff"
         self.error_text_color = "#ff5252"
 
+
+>>>>>>> refs/remotes/origin/main
 # Custom CSS for exact layout and styling
 custom_css = """
 body {