johann22 commited on
Commit
f7e6ab1
·
1 Parent(s): e3d392f

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +3 -3
app_dialogue.py CHANGED
@@ -64,8 +64,8 @@ STOP_SUSPECT_LIST = []
64
 
65
  GRADIO_LINK = "https://johann22-idefics-playground.hf.space"
66
  API_TOKEN = os.getenv("HF_AUTH_TOK")
67
- #IDEFICS_LOGO = "https://huggingface.co/spaces/johann22/idefics_playground/resolve/main/IDEFICS_logo.png"
68
- IDEFICS_LOGO = "IDEFICS_logo.png"
69
 
70
  PROCESSOR = AutoProcessor.from_pretrained(
71
  "HuggingFaceM4/idefics-9b-instruct",
@@ -578,7 +578,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
578
  Same as `model_inference` but in greedy mode and with the 80b-instruct.
579
  Specifically for pre-computing the default examples.
580
  """
581
- model_selector="HuggingFaceM4/idefics-80b-instruct"
582
  user_prompt_str=message
583
  chat_history=[]
584
  max_new_tokens=512
 
64
 
65
  GRADIO_LINK = "https://johann22-idefics-playground.hf.space"
66
  API_TOKEN = os.getenv("HF_AUTH_TOK")
67
+ IDEFICS_LOGO = "https://huggingface.co/spaces/johann22/idefics_playground/resolve/main/IDEFICS_logo.png"
68
+ #IDEFICS_LOGO = "IDEFICS_logo.png"
69
 
70
  PROCESSOR = AutoProcessor.from_pretrained(
71
  "HuggingFaceM4/idefics-9b-instruct",
 
578
  Same as `model_inference` but in greedy mode and with the 80b-instruct.
579
  Specifically for pre-computing the default examples.
580
  """
581
+ model_selector="HuggingFaceM4/idefics-9b-instruct"
582
  user_prompt_str=message
583
  chat_history=[]
584
  max_new_tokens=512