Update app.py

app.py

@@ -32,10 +32,10 @@ def generation(tokenizer, model, content):
 
     seek_token_ids = tokenizer.encode('= " +')[1:]
 
-    full_output = lm.generate(code, generate=10, do_sample=False)
+    full_output = model.generate(code, generate=6, do_sample=False)
 
     def next_words(code, position, seek_token_ids):
-        op_model = lm.generate(code, generate=1, do_sample=False)
+        op_model = model.generate(code, generate=1, do_sample=False)
         hidden_states = op_model.hidden_states
         layer_no = len(hidden_states) - 1
         h = hidden_states[-1]