zliang commited on
Commit
b657107
1 Parent(s): 6e1a1ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,8 +1,6 @@
1
 
2
 
3
- # Load the trained model
4
- model = YOLO("best.pt")
5
- openai_api_key = os.environ.get("openai_api_key")
6
 
7
  import os
8
  import io
@@ -26,6 +24,8 @@ from streamlit_chat import message
26
 
27
  # Load the trained model
28
 
 
 
29
 
30
  # Define the class indices for figures, tables, and text
31
  figure_class_index = 4
 
1
 
2
 
3
+
 
 
4
 
5
  import os
6
  import io
 
24
 
25
  # Load the trained model
26
 
27
+ model = YOLO("best.pt")
28
+ openai_api_key = os.environ.get("openai_api_key")
29
 
30
  # Define the class indices for figures, tables, and text
31
  figure_class_index = 4