ngebodh commited on
Commit
094183d
1 Parent(s): 4aaa844

Updated and added more models

Browse files
Files changed (1) hide show
  1. app.py +14 -3
app.py CHANGED
@@ -22,7 +22,9 @@ client = OpenAI(
22
  #Create supported models
23
  model_links ={
24
  "Mistral":"mistralai/Mistral-7B-Instruct-v0.2",
25
- "Gemma":"google/gemma-7b-it",
 
 
26
  # "Llama-2":"meta-llama/Llama-2-7b-chat-hf"
27
 
28
  }
@@ -33,16 +35,25 @@ model_info ={
33
  {'description':"""The Mistral model is a **Large Language Model (LLM)** that's able to have question and answer interactions.\n \
34
  \nIt was created by the [**Mistral AI**](https://mistral.ai/news/announcing-mistral-7b/) team as has over **7 billion parameters.** \n""",
35
  'logo':'https://mistral.ai/images/logo_hubc88c4ece131b91c7cb753f40e9e1cc5_2589_256x0_resize_q97_h2_lanczos_3.webp'},
36
- "Gemma":
37
  {'description':"""The Gemma model is a **Large Language Model (LLM)** that's able to have question and answer interactions.\n \
38
  \nIt was created by the [**Google's AI Team**](https://blog.google/technology/developers/gemma-open-models/) team as has over **7 billion parameters.** \n""",
39
  'logo':'https://pbs.twimg.com/media/GG3sJg7X0AEaNIq.jpg'},
 
 
 
 
 
 
 
 
 
 
40
 
41
  }
42
 
43
 
44
  # Define the available models
45
- # models = ["Mistral", "Gemma"]
46
  models =[key for key in model_links.keys()]
47
 
48
  # Create the sidebar with the dropdown for model selection
 
22
  #Create supported models
23
  model_links ={
24
  "Mistral":"mistralai/Mistral-7B-Instruct-v0.2",
25
+ "Gemma-7B":"google/gemma-7b-it",
26
+ "Gemma-2B":"google/gemma-2b-it",
27
+ "Gemma-Zephyr":"HuggingFaceH4/zephyr-7b-gemma-v0.1",
28
  # "Llama-2":"meta-llama/Llama-2-7b-chat-hf"
29
 
30
  }
 
35
  {'description':"""The Mistral model is a **Large Language Model (LLM)** that's able to have question and answer interactions.\n \
36
  \nIt was created by the [**Mistral AI**](https://mistral.ai/news/announcing-mistral-7b/) team as has over **7 billion parameters.** \n""",
37
  'logo':'https://mistral.ai/images/logo_hubc88c4ece131b91c7cb753f40e9e1cc5_2589_256x0_resize_q97_h2_lanczos_3.webp'},
38
+ "Gemma-7B":
39
  {'description':"""The Gemma model is a **Large Language Model (LLM)** that's able to have question and answer interactions.\n \
40
  \nIt was created by the [**Google's AI Team**](https://blog.google/technology/developers/gemma-open-models/) team as has over **7 billion parameters.** \n""",
41
  'logo':'https://pbs.twimg.com/media/GG3sJg7X0AEaNIq.jpg'},
42
+ "Gemma-2B":
43
+ {'description':"""The Gemma model is a **Large Language Model (LLM)** that's able to have question and answer interactions.\n \
44
+ \nIt was created by the [**Google's AI Team**](https://blog.google/technology/developers/gemma-open-models/) team as has over **2 billion parameters.** \n""",
45
+ 'logo':'https://pbs.twimg.com/media/GG3sJg7X0AEaNIq.jpg'},
46
+ "Gemma-Zephyr":
47
+ {'description':"""The Gemma model is a **Large Language Model (LLM)** that's able to have question and answer interactions.\n \
48
+ \nFrom Huggingface: Zephyr is a series of language models that are trained to act as helpful assistants. \
49
+ Zephyr 7B Gemma is the third model in the series, and is a fine-tuned version of google/gemma-7b \
50
+ that was trained on on a mix of publicly available, synthetic datasets using Direct Preference Optimization (DPO)\n""",
51
+ 'logo':'https://huggingface.co/HuggingFaceH4/zephyr-7b-gemma-v0.1/resolve/main/thumbnail.png'},
52
 
53
  }
54
 
55
 
56
  # Define the available models
 
57
  models =[key for key in model_links.keys()]
58
 
59
  # Create the sidebar with the dropdown for model selection