Create app.py

app.py

@@ -0,0 +1,14 @@
+import gradio as gr #webinterface
+from transformers import AutoModelForCasualLM,Auto Tokenizer # for loading the moel and making the inputs into tokens
+model_name="Salesforce/codegen-350M-multi"
+
+#initialize the tokenizer and model
+tokenizer=Auto Tokenizer.from_pretrained(model_name)
+model=AutoModelForCasualLM.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)
+
+    generated_code=tokenizer.decode(outputs[0],skip_special_tokens=True)
+    return generated_code