Update app.py
Browse files
app.py
CHANGED
|
@@ -34,7 +34,9 @@ def respond(
|
|
| 34 |
|
| 35 |
demo = gr.Interface(
|
| 36 |
respond,
|
| 37 |
-
inputs=[
|
|
|
|
|
|
|
| 38 |
additional_inputs=[
|
| 39 |
gr.Textbox(value="You are a movie review classifier. You respond to given movie reviews with a predicted star rating 0-5 (inclusive) for that review with no explanation.", label="System message"),
|
| 40 |
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
|
@@ -47,7 +49,9 @@ demo = gr.Interface(
|
|
| 47 |
label="Top-p (nucleus sampling)",
|
| 48 |
),
|
| 49 |
],
|
| 50 |
-
outputs=[
|
|
|
|
|
|
|
| 51 |
)
|
| 52 |
|
| 53 |
|
|
|
|
| 34 |
|
| 35 |
demo = gr.Interface(
|
| 36 |
respond,
|
| 37 |
+
inputs=[
|
| 38 |
+
gr.Textbox(label="Movie review") # Set label for the input
|
| 39 |
+
],
|
| 40 |
additional_inputs=[
|
| 41 |
gr.Textbox(value="You are a movie review classifier. You respond to given movie reviews with a predicted star rating 0-5 (inclusive) for that review with no explanation.", label="System message"),
|
| 42 |
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
|
|
|
| 49 |
label="Top-p (nucleus sampling)",
|
| 50 |
),
|
| 51 |
],
|
| 52 |
+
outputs=[
|
| 53 |
+
gr.Textbox(label="Star rating") # Set label for the output
|
| 54 |
+
],
|
| 55 |
)
|
| 56 |
|
| 57 |
|