Update app.py

app.py

@@ -16,6 +16,20 @@ NUM_THREADS = 2
 model = whisper.load_model("small")
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
+
+def parse_codeblock(text):
+    lines = text.split("\n")
+    for i, line in enumerate(lines):
+        if "```" in line:
+            if line != "```":
+                lines[i] = f'<pre><code class="{lines[i][3:]}">'
+            else:
+                lines[i] = '</code></pre>'
+        else:
+            if i > 0:
+                lines[i] = "<br/>" + line.replace("<", "&lt;").replace(">", "&gt;")
+    return "".join(lines)
+    
 def inference(audio):
     audio = whisper.load_audio(audio)
     print("loading finished")
@@ -137,6 +151,7 @@ def gpt_predict(inputs, request:gr.Request=gr.State([]), top_p = 1, temperature
                 # decode each line as response data is in bytes
                 if len(chunk) > 12 and "content" in json.loads(chunk[6:])['choices'][0]['delta']:
                     partial_words = partial_words + json.loads(chunk[6:])['choices'][0]["delta"]["content"]
+                    print(partial_words)
                     if token_counter == 0:
                         history.append(" " + partial_words)
                     else: