Grandediw
commited on
Commit
·
50cc010
1
Parent(s):
1322782
Update Interface
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ def infer(prompt):
|
|
| 36 |
# Placeholder return, modify based on your specific model task
|
| 37 |
return outputs.last_hidden_state.mean(dim=1).cpu().detach().numpy()
|
| 38 |
|
| 39 |
-
# Gradio interface
|
| 40 |
with gr.Blocks() as demo:
|
| 41 |
gr.Markdown("## LoRA Model Inference")
|
| 42 |
|
|
|
|
| 36 |
# Placeholder return, modify based on your specific model task
|
| 37 |
return outputs.last_hidden_state.mean(dim=1).cpu().detach().numpy()
|
| 38 |
|
| 39 |
+
# Gradio interface Update
|
| 40 |
with gr.Blocks() as demo:
|
| 41 |
gr.Markdown("## LoRA Model Inference")
|
| 42 |
|