TeamGreenEdifai commited on
Commit
433d5f1
·
verified ·
1 Parent(s): f24285f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -34,7 +34,9 @@ def respond(
34
 
35
  demo = gr.Interface(
36
  respond,
37
- inputs=["text"],
 
 
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=["text"],
 
 
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