david167 commited on
Commit
8b4bf36
·
1 Parent(s): 0c60639

Fix: Add missing generated_text variable definition

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -115,7 +115,7 @@ def chat_response(message: str, history: List[List[str]], temperature: float) ->
115
  )
116
 
117
  # Decode response
118
- # TEMPORARY: Show complete raw output to debug clipping
119
  response = f"""=== RAW MODEL OUTPUT ===
120
  {generated_text}
121
  === END RAW OUTPUT ===
 
115
  )
116
 
117
  # Decode response
118
+ generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True) # TEMPORARY: Show complete raw output to debug clipping
119
  response = f"""=== RAW MODEL OUTPUT ===
120
  {generated_text}
121
  === END RAW OUTPUT ===