Kvikontent commited on
Commit
8493dae
·
1 Parent(s): 76a650f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -16,8 +16,11 @@ def generate_response(text):
16
  # Get the last hidden states from the model output
17
  last_hidden_states = outputs.last_hidden_state
18
 
 
 
 
19
  # Return last hidden states as output
20
- return last_hidden_states
21
 
22
  iface = gr.Interface(
23
  fn=generate_response,
 
16
  # Get the last hidden states from the model output
17
  last_hidden_states = outputs.last_hidden_state
18
 
19
+ # Convert tensor to a human-readable format
20
+ output_text = "\n".join([str(token) for token in last_hidden_states[0][0]])
21
+
22
  # Return last hidden states as output
23
+ return output_text
24
 
25
  iface = gr.Interface(
26
  fn=generate_response,