ysharma HF Staff commited on
Commit
9790edd
·
1 Parent(s): 847666f

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +2 -4
app_dialogue.py CHANGED
@@ -21,11 +21,9 @@ MODELS = [ # TODO uncomment
21
  ]
22
 
23
  API_PATHS = { # TODO uncomment + correct endpoint
24
- "HuggingFaceM4/idefics-9b-instruct": "https://api-inference.huggingface.co/models/HuggingFaceM4/idefics-9b-instruct",
25
  #"HuggingFaceM4/idefics-9b-instruct": os.getenv("TEMP_DEPLOY_ENDPOINT"),
26
- "HuggingFaceM4/idefics-80b-instruct": (
27
- "https://api-inference.huggingface.co/models/HuggingFaceM4/idefics-80b-instruct"
28
- ),
29
  }
30
 
31
  SYSTEM_PROMPT = [
 
21
  ]
22
 
23
  API_PATHS = { # TODO uncomment + correct endpoint
24
+ "HuggingFaceM4/idefics-9b-instruct": os.getenv("9B"),
25
  #"HuggingFaceM4/idefics-9b-instruct": os.getenv("TEMP_DEPLOY_ENDPOINT"),
26
+ "HuggingFaceM4/idefics-80b-instruct": os.getenv("80B"),
 
 
27
  }
28
 
29
  SYSTEM_PROMPT = [