noelstan99 commited on
Commit
66ce146
·
1 Parent(s): d7a1db1

fix app.py - page config-

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -2,6 +2,13 @@ import streamlit as st
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
 
 
 
 
 
 
 
5
  # Load model and tokenizer
6
  @st.cache_resource
7
  def load_model():
@@ -13,12 +20,6 @@ def load_model():
13
  tokenizer, model = load_model()
14
 
15
  # App UI
16
- st.set_page_config(
17
- page_title="EdStem Chatbot Demo",
18
- page_icon="🤖",
19
- layout="wide",
20
- )
21
-
22
  st.title("🤖 EdStem Chatbot Demo")
23
  st.markdown(
24
  """
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
5
+ # Set page configuration - MUST be the first Streamlit command
6
+ st.set_page_config(
7
+ page_title="EdStem Chatbot Demo",
8
+ page_icon="🤖",
9
+ layout="wide",
10
+ )
11
+
12
  # Load model and tokenizer
13
  @st.cache_resource
14
  def load_model():
 
20
  tokenizer, model = load_model()
21
 
22
  # App UI
 
 
 
 
 
 
23
  st.title("🤖 EdStem Chatbot Demo")
24
  st.markdown(
25
  """