mohammedriza-rahman commited on
Commit
dfbd341
·
verified ·
1 Parent(s): 5a1e477

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py CHANGED
@@ -1,6 +1,21 @@
1
  import streamlit as st
2
  from transformers import AutoModel, AutoTokenizer
3
  import torch
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
 
5
  # Page config
6
  st.set_page_config(
 
1
  import streamlit as st
2
  from transformers import AutoModel, AutoTokenizer
3
  import torch
4
+ import sys
5
+ import subprocess
6
+ import streamlit as st
7
+
8
+ st.write(f"Python version: {sys.version}")
9
+ st.write(f"Pip version: {subprocess.check_output(['pip', '--version']).decode()}")
10
+
11
+ try:
12
+ from transformers import AutoModel, AutoTokenizer
13
+ st.success("Transformers package is installed!")
14
+ except ImportError:
15
+ st.error("Transformers package is not installed")
16
+ st.write("Installed packages:")
17
+ st.code(subprocess.check_output(['pip', 'list']).decode())
18
+
19
 
20
  # Page config
21
  st.set_page_config(