Update app.py

app.py

@@ -1,9 +1,12 @@
 fine_tuned_model = "ashioyajotham/results"
 
 def generate_code(input_text):
+    inputs = tokenizer.encode(input_text, return_tensors='pt')
     # Use your fine-tuned model to generate code here
-    generated_code = fine_tuned_model.generate(input_text)
-    return generated_code
+    generated_code = fine_tuned_model.generate(input_text, max_length=100, temperature = .7, do_sample=True, pad_token_id=tokenizer.eos_token_id)
+    completion = tokenizer.decode(generated_code[0])
+
+    return completion
 
 import gradio as gr