aldohenrique commited on
Commit
43e4431
·
verified ·
1 Parent(s): 76b8dca

Update ai_logic.py

Browse files
Files changed (1) hide show
  1. ai_logic.py +2 -0
ai_logic.py CHANGED
@@ -61,6 +61,8 @@ class HuggingFaceProvider(LLMProvider):
61
  self.token = token
62
  self.headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
63
  self.models = [
 
 
64
  ("Phi-3 Mini (Mais rápido)", "microsoft/Phi-3-mini-4k-instruct"),
65
  ("Zephyr 7B (Meio Termo)", "HuggingFaceH4/zephyr-7b-beta"),
66
  ("Mistral-7B", "mistralai/Mistral-7B-Instruct-v0.3"),
 
61
  self.token = token
62
  self.headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
63
  self.models = [
64
+ ("IBM 2B","ibm-granite/granite-speech-3.3-2b"),
65
+ ("IBM 8B","ibm-granite/granite-speech-3.3-8b"),
66
  ("Phi-3 Mini (Mais rápido)", "microsoft/Phi-3-mini-4k-instruct"),
67
  ("Zephyr 7B (Meio Termo)", "HuggingFaceH4/zephyr-7b-beta"),
68
  ("Mistral-7B", "mistralai/Mistral-7B-Instruct-v0.3"),