Spaces:

mkurman
/

Llama-3.2-MEDIT-3B-o1

Running

mkurman commited on Jan 5

Commit

bce3ed1

1 Parent(s): 7f71d93

small fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,8 +47,11 @@ def respond(
         if token in ['<Thought>', '<|python_tag|>']:
             text = '## Thinking:\n\n' + text
-        if token == '<Output>' and counter > 0:
-            text = '## Output:\n\n' + text
         if token == '</Output>':
             yield None

         if token in ['<Thought>', '<|python_tag|>']:
             text = '## Thinking:\n\n' + text
+        if token == '<Output>':
+            if counter > 0:
+                text = '## Output:\n\n' + text
+            else:
+                text = '## Thinking:\n\n' + text
         if token == '</Output>':
             yield None