Update app.py

app.py

@@ -8,7 +8,7 @@ model=AutoModelForCausalLM.from_pretrained(model_name)
 
 def generate_code(prompt, max_length=100, temperature=0.7, top_p=0.95):
     inputs=tokenizer(prompt,return_tensors='pt') 
-    output=model.generate(**inputs, max_length=max_length, temperature=temperature, top_p=top_p, do_sample=True) #input: input_id, weight_number
+    outputs=model.generate(**inputs, max_length=max_length, temperature=temperature, top_p=top_p, do_sample=True) #input: input_id, weight_number
 
     generated_code=tokenizer.decode(outputs[0],skip_special_tokens=True)
     return generated_code