prasanna kumar commited on
Commit
43be51d
1 Parent(s): eb89467

added gemma model support

Browse files
app.py CHANGED
@@ -6,7 +6,7 @@ model_path = "models/"
6
  import gradio as gr
7
 
8
  # Available models
9
- MODELS = ["Meta-Llama-3.1-8B"]
10
 
11
  def process_input(input_type, input_value, model_name):
12
  # Initialize tokenizer
 
6
  import gradio as gr
7
 
8
  # Available models
9
+ MODELS = ["Meta-Llama-3.1-8B","gemma-2b"]
10
 
11
  def process_input(input_type, input_value, model_name):
12
  # Initialize tokenizer
models/gemma-2b/model.safetensors.index.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b32948684e244670c6d7c9a9dc723bce5d4b2b34319879c9c1844a0cbf1a15a
3
+ size 13489
models/gemma-2b/special_tokens_map.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:baec30ea10906f16adb8c18af7a34023002c1746542612b8b41c9f09e1351351
3
+ size 636
models/gemma-2b/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c15eb04bc5ad609fb26533e8525302c5640a945e5f67f65b7c849900acda7d99
3
+ size 17518497
models/gemma-2b/tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
3
+ size 4241003
models/gemma-2b/tokenizer_config.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1cc6e515e5900205fcc07b62fd16503bbff92b9c2c22bed56fd64400402fbb5
3
+ size 33556