Francisco Zanartu commited on
Commit
6e311b6
1 Parent(s): f8b8183

update requirements and card model endpoint

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. utils/core.py +1 -1
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  langchain==0.1.12
2
  duckduckgo-search==5.1.0
 
3
 
4
  # google-generativeai==0.3.2
5
  # langchain==0.1.9
 
1
  langchain==0.1.12
2
  duckduckgo-search==5.1.0
3
+ transformers==4.38.2
4
 
5
  # google-generativeai==0.3.2
6
  # langchain==0.1.9
utils/core.py CHANGED
@@ -145,7 +145,7 @@ class HamburgerStyle:
145
 
146
  ## FACT: Concluding
147
  cards_label = self.endpoint_query(
148
- model=self.card_model, payload=misinformation
149
  )[0][0].get("label")
150
  # 1 get all claims with same label from FEVER dataset
151
  claims = self.get_fever_claims(cards_label) # TODO
 
145
 
146
  ## FACT: Concluding
147
  cards_label = self.endpoint_query(
148
+ model=self.taxonomy_cards, payload=misinformation
149
  )[0][0].get("label")
150
  # 1 get all claims with same label from FEVER dataset
151
  claims = self.get_fever_claims(cards_label) # TODO