Spaces:
Configuration error
Configuration error
Update app.py
Browse files
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 (
|
| 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="/")
|