Update app.py

app.py

@@ -7,18 +7,19 @@ tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model, trust_remote_code=Tr
 # Set the padding token to be the same as the end-of-sequence token
 tokenizer.pad_token = tokenizer.eos_token
 
-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, max_length=100, temperature = .7, do_sample=True, pad_token_id=tokenizer.eos_token_id)
-    completion = tokenizer.decode(generated_code[0])
+prompt = "Generate a Python script to add prime numbers between one and ten"
 
-    return completion
+inputs = tokenizer.encode(prompt, return_tensors='pt')
 
+outputs = model.generate(inputs, max_length=100, temperature = .7, do_sample=True, pad_token_id=tokenizer.eos_token_id)
+
+completion = tokenizer.decode(outputs[0])
+
+#print(completion)
 import gradio as gr
 
 iface = gr.Interface(
-    fn=generate_code,
+    fn=completion,
     inputs="text",
     outputs="text",
     title="Code Generation App",