Update app.py

app.py

@@ -23,7 +23,7 @@ outputs = model(**inputs) #, labels=inputs["input_ids"])
 
 last_hidden_states = outputs.last_hidden_state
 
-output = last_hidden_states
+output = last_hidden_states[0][0]
 
 # decoded = tokenizer.decode(output[0][0]) # [0][0]
 # print(decoded)