johann22 commited on
Commit
cda4eb3
·
1 Parent(s): d8578f4

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +3 -1
app_dialogue.py CHANGED
@@ -594,13 +594,15 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
594
  current_image=image,
595
  history=chat_history,
596
  )
 
597
 
 
598
  client_endpoint = API_PATHS[model_selector]
599
  client = Client(
600
  base_url=client_endpoint,
601
  headers={"x-use-cache": "0", "Authorization": f"Bearer {API_TOKEN}"},
602
  timeout=240, # Generous time out just in case because we are in greedy. All examples should be computed in less than 30secs with the 80b-instruct.
603
- )
604
 
605
  # Common parameters to all decoding strategies
606
  # This documentation is useful to read: https://huggingface.co/docs/transformers/main/en/generation_strategies
 
594
  current_image=image,
595
  history=chat_history,
596
  )
597
+ client = InferenceClient("HuggingFaceM4/idefics-80b-instruct")
598
 
599
+ '''
600
  client_endpoint = API_PATHS[model_selector]
601
  client = Client(
602
  base_url=client_endpoint,
603
  headers={"x-use-cache": "0", "Authorization": f"Bearer {API_TOKEN}"},
604
  timeout=240, # Generous time out just in case because we are in greedy. All examples should be computed in less than 30secs with the 80b-instruct.
605
+ )'''
606
 
607
  # Common parameters to all decoding strategies
608
  # This documentation is useful to read: https://huggingface.co/docs/transformers/main/en/generation_strategies