Omnibus commited on
Commit
3c03c99
1 Parent(s): c803888

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +4 -4
app_dialogue.py CHANGED
@@ -18,8 +18,8 @@ from transformers import AutoProcessor
18
 
19
  MODELS = [
20
  # "HuggingFaceM4/idefics-9b-instruct",
21
- "tiiuae/falcon-180B",
22
- # "HuggingFaceM4/idefics-80b-instruct",
23
  ]
24
 
25
  API_PATHS = {
@@ -68,10 +68,10 @@ GRADIO_LINK = "https://omnibus-idefics-playground.hf.space"
68
  API_TOKEN = os.getenv("HF_AUTH_TOKEN")
69
  IDEFICS_LOGO = "https://huggingface.co/spaces/HuggingFaceM4/idefics_playground/resolve/main/IDEFICS_logo.png"
70
 
71
- # "HuggingFaceM4/idefics-9b-instruct",
72
 
73
  PROCESSOR = AutoProcessor.from_pretrained(
74
- "tiiuae/falcon-180B",
75
  token=API_TOKEN,
76
  )
77
 
 
18
 
19
  MODELS = [
20
  # "HuggingFaceM4/idefics-9b-instruct",
21
+ #"tiiuae/falcon-180B",
22
+ "HuggingFaceM4/idefics-80b-instruct",
23
  ]
24
 
25
  API_PATHS = {
 
68
  API_TOKEN = os.getenv("HF_AUTH_TOKEN")
69
  IDEFICS_LOGO = "https://huggingface.co/spaces/HuggingFaceM4/idefics_playground/resolve/main/IDEFICS_logo.png"
70
 
71
+
72
 
73
  PROCESSOR = AutoProcessor.from_pretrained(
74
+ "HuggingFaceM4/idefics-9b-instruct",
75
  token=API_TOKEN,
76
  )
77