Update app.py

app.py

@@ -13,7 +13,7 @@ config = PeftConfig.from_pretrained("jonathanjordan21/mos-mamba-6x130m-trainer")
 
 tokenizer = AutoTokenizer.from_pretrained("jonathanjordan21/mos-mamba-6x130m-trainer", trust_remote_code=True)
 
-streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
+# streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 
 model = AutoModelForCausalLM.from_pretrained(
     "jonathanjordan21/mos-mamba-6x130m-trainer",
@@ -59,7 +59,9 @@ def invoke(
 
     res = tokenizer.batch_decode(response)
 
-    return res
+    print(res)
+
+    return res[0]
     # yield res