import gradio as gr
import os 
import numpy as np 

import keras_nlp
preprocessor = keras_nlp.models.GemmaPreprocessor.from_preset(
    "gemma_2b_en", sequence_length=64
)
model = keras_nlp.models.GemmaBackbone.from_preset("gemma_2b_en")


def greet(name):
  x = preprocessor(name)
  lab = model(x)
  return lab

iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()