Nav772 commited on
Commit
04fe45a
·
verified ·
1 Parent(s): 11cabcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,9 +12,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
12
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
13
  class BasicAgent:
14
  def __init__(self):
15
- print("FLAN-T5 API Agent initialized.")
16
  self.api_token = os.getenv("HF_API_TOKEN")
17
- self.api_url = "https://api-inference.huggingface.co/models/google/flan-t5-small"
18
  self.headers = {
19
  "Authorization": f"Bearer {self.api_token}",
20
  "Content-Type": "application/json"
 
12
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
13
  class BasicAgent:
14
  def __init__(self):
15
+ print("Mistral Instruct Agent initialized.")
16
  self.api_token = os.getenv("HF_API_TOKEN")
17
+ self.api_url = "https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.1"
18
  self.headers = {
19
  "Authorization": f"Bearer {self.api_token}",
20
  "Content-Type": "application/json"