AlexandreScriptsMT commited on
Commit
0ab3859
·
verified ·
1 Parent(s): 6276f44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def generate(prompt):
32
  out = generator(prompt, max_new_tokens=64, do_sample=False)
33
  return out[0]["generated_text"]
34
 
35
- demo = gr.Interface(fn=generate, inputs=gr.Textbox(lines=4, label="Prompt"), outputs="text", title="Gemma (Space CPU)")
36
 
37
  app = FastAPI()
38
  app = gr.mount_gradio_app(app, demo, path="/")
 
32
  out = generator(prompt, max_new_tokens=64, do_sample=False)
33
  return out[0]["generated_text"]
34
 
35
+ demo = gr.Interface(fn=generate, inputs=gr.Textbox(lines=4, label="Prompt"), outputs="text", title="Gemma-4-E2B (CPU)")
36
 
37
  app = FastAPI()
38
  app = gr.mount_gradio_app(app, demo, path="/")