tensorgirl commited on
Commit
8d40135
·
verified ·
1 Parent(s): ea2f8b8

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +25 -0
  2. requirements.txt +4 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import os
3
+ import numpy as np
4
+ import librosa
5
+ import tensorflow as tf
6
+ from tensorflow import keras
7
+ from tensorflow.keras import layers
8
+ from transformers import AutoFeatureExtractor
9
+ from sklearnex import patch_sklearn, unpatch_sklearn
10
+ patch_sklearn()
11
+ import xgboost as xgb
12
+ import keras_nlp
13
+ preprocessor = keras_nlp.models.GemmaPreprocessor.from_preset(
14
+ "gemma_2b_en", sequence_length=64
15
+ )
16
+ model = keras_nlp.models.GemmaBackbone.from_preset("gemma_2b_en")
17
+
18
+
19
+ def greet(name):
20
+ x = preprocessor(name)
21
+ lab = model(x)
22
+ return lab
23
+
24
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
25
+ iface.launch()
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ keras-nlp
2
+ tensorflow
3
+ sklearnex
4
+ keras