SusiePHaltmann commited on
Commit
3fed8da
·
1 Parent(s): a596f98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -3,6 +3,17 @@ import streamlit as st
3
  @st.cache
4
 
5
  import sys
 
 
 
 
 
 
 
 
 
 
 
6
  sys.setrecursionlimit(100000)
7
  def load_model():
8
  model = Megatron()
 
3
  @st.cache
4
 
5
  import sys
6
+ if len(sys.argv) != 2:
7
+ print("Please provide a filename")
8
+ sys.exit(1)
9
+
10
+ filename = sys.argv[1]
11
+
12
+ # Your code goes here
13
+
14
+
15
+ if __name__ == "__main__":
16
+ main()
17
  sys.setrecursionlimit(100000)
18
  def load_model():
19
  model = Megatron()