Update app.py

app.py

@@ -1,21 +1,30 @@
-import streamlit as st
+# import streamlit as st
 from transformers import AutoModel, AutoTokenizer
 import torch
-import sys
-import subprocess
 import streamlit as st
+import subprocess
+import sys
+
+st.title("Package Installation Test")
 
+# Display Python version
 st.write(f"Python version: {sys.version}")
-st.write(f"Pip version: {subprocess.check_output(['pip', '--version']).decode()}")
 
+# Try to install transformers
 try:
-    from transformers import AutoModel, AutoTokenizer
-    st.success("Transformers package is installed!")
-except ImportError:
-    st.error("Transformers package is not installed")
-    st.write("Installed packages:")
-    st.code(subprocess.check_output(['pip', 'list']).decode())
+    st.write("Attempting to install transformers...")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers"])
+    st.success("Transformers package installed successfully!")
+except Exception as e:
+    st.error(f"Error installing transformers: {str(e)}")
 
+# List installed packages
+st.write("Installed packages:")
+try:
+    installed_packages = subprocess.check_output([sys.executable, "-m", "pip", "list"]).decode()
+    st.code(installed_packages)
+except Exception as e:
+    st.error(f"Error listing packages: {str(e)}")
 
 # Page config
 st.set_page_config(