Omnibus commited on
Commit
c803888
1 Parent(s): f2c5d9b

Update app_dialogue.py

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