yaswanthkocherla commited on
Commit
69ef173
·
verified ·
1 Parent(s): 2ac1269

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,10 +1,13 @@
1
  import streamlit as st
2
- from transformers import pipeline
 
 
 
3
 
4
  st.title("🏙️ Sustainable Smart City Assistant")
5
  st.write("Ask sustainability questions")
6
 
7
- qa_pipeline = pipeline("text-generation", model="open-llm-leaderboard/ibm-granite__granite-3.0-2b-instruct-details")
8
 
9
  query = st.text_input("Enter your question")
10
  if query:
 
1
  import streamlit as st
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM
3
+
4
+ tokenizer = AutoTokenizer.from_pretrained("ibm-granite/granite-3.3-8b-instruct")
5
+ model = AutoModelForCausalLM.from_pretrained("ibm-granite/granite-3.3-8b-instruct")
6
 
7
  st.title("🏙️ Sustainable Smart City Assistant")
8
  st.write("Ask sustainability questions")
9
 
10
+ qa_pipeline = pipeline("text-generation", model="ibm-granite/granite-3.3-8b-instruct")
11
 
12
  query = st.text_input("Enter your question")
13
  if query: