Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from streamlit_ace import st_ace
-from streamlit_jupyter import st_jupyter
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 import os
 import subprocess
@@ -12,6 +11,11 @@ import torch
 from huggingface_hub import hf_hub_url, cached_download, HfApi
 import re
 from typing import List, Dict
+import streamlit as st
+from streamlit_jupyter import StreamlitPatcher, tqdm
+
+StreamlitPatcher().jupyter()  # This patches Streamlit to work in Jupyter
+
 
 # Access Hugging Face API key from secrets
 hf_token = st.secrets["hf_token"]