fix output

app.py

@@ -122,8 +122,8 @@ def run():
 
     if submit_button:
         
-        generate_solution(model, tokenizer, question, starter_code, temperature, num_beams)
-        st.code(tmp, language="python")
+        output = generate_solution(model, tokenizer, question, starter_code, temperature, num_beams)
+        st.code(output, language="python")
     
 
 if __name__=="__main__":