CosmoAI commited on
Commit
dfc3c17
·
1 Parent(s): 77b20da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -2,9 +2,10 @@ from bardapi import Bard
2
  import os
3
  import streamlit as st
4
  from transformers import pipeline
 
5
 
6
  intent = pipeline(model="facebook/bart-large-mnli")
7
- fetchvalue = pipeline(model="Jean-Baptiste/camembert-ner", aggregation_strategy="simple")
8
 
9
  bardkey = os.environ.get("BARD_API_KEY")
10
 
@@ -12,7 +13,7 @@ bardkey = os.environ.get("BARD_API_KEY")
12
  if query := st.chat_input("Hi, how can I help you"):
13
  usrintent = intent(query, candidate_labels=["Reminder", "General Conversation"])
14
  if usrintent["labels"][0] == "Reminder":
15
- values = fetchvalue(query)
16
  with st.chat_message("assistant"):
17
  st.write(values)
18
  elif usrintent["labels"][0] == "General Conversation":
 
2
  import os
3
  import streamlit as st
4
  from transformers import pipeline
5
+ from getvalues import getValues
6
 
7
  intent = pipeline(model="facebook/bart-large-mnli")
8
+ # fetchvalue = pipeline(model="Jean-Baptiste/camembert-ner", aggregation_strategy="simple")
9
 
10
  bardkey = os.environ.get("BARD_API_KEY")
11
 
 
13
  if query := st.chat_input("Hi, how can I help you"):
14
  usrintent = intent(query, candidate_labels=["Reminder", "General Conversation"])
15
  if usrintent["labels"][0] == "Reminder":
16
+ values = getValues(query)
17
  with st.chat_message("assistant"):
18
  st.write(values)
19
  elif usrintent["labels"][0] == "General Conversation":