Update app.py

app.py

@@ -21,12 +21,12 @@ def generate_output(prompt,maxtokens=20):
     toc = time.time()
     raw_data = data.decode("utf-8")
     json_data = json.loads(raw_data)
-    return json_data["prediction"], round(toc-tic,3)
+    return json_data["prediction"], str(round(toc-tic,3))+"s"
 
 input_text = gr.inputs.Textbox(label="Enter Prompt")
 slider = gr.inputs.Slider(minimum=1, maximum=75, step=1, default=20, label="Max New Tokens")
 output_text = gr.outputs.Textbox(label="Response")
-time_text = gr.outputs.Label(label="Time Taken (s)")
+time_text = gr.outputs.Label(label="Time Taken")
 
 title = "SPR - Cnvrg.io | Serving"
 description = "Interact with the endpoint - Falcon 7B (Instruct)"