Update README.md

README.md

@@ -37,13 +37,13 @@ There was great political strife in the air, and tensions were high. People rush
                 num_beams=4
             )
 
-    text = '[User]: What's the best food in Hanoi?''
-    input_ids = st.session_state.tokenizer(text, return_tensors='pt')['input_ids'].to('cuda')
-    with torch.no_grad():
-        output = st.session_state['model'].generate(input_ids=input_ids, generation_config=st.session_state.gen_config,return_dict_in_generate=True, output_scores=True,max_new_tokens=256)
-        s = output.sequences[0]
-        output = st.session_state.tokenizer.decode(s)
-        print('Raw:',output)
+        text = '[User]: What's the best food in Hanoi?''
+        input_ids = st.session_state.tokenizer(text, return_tensors='pt')['input_ids'].to('cuda')
+        with torch.no_grad():
+            output = st.session_state['model'].generate(input_ids=input_ids, generation_config=st.session_state.gen_config,return_dict_in_generate=True, output_scores=True,max_new_tokens=256)
+            s = output.sequences[0]
+            output = st.session_state.tokenizer.decode(s)
+            print('Raw:',output)
         
 ```