Update app.py

Updated the model.

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load GraphCodeBERT model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("microsoft/graphcodebert-base")
-model = AutoModel.from_pretrained("microsoft/graphcodebert-base")
+model = AutoModelForCausalLM.from_pretrained("microsoft/graphcodebert-base")
 
 # Define input and output interfaces
 input = gr.Textbox(lines=5, label="Input")
@@ -21,5 +21,16 @@ def use_graphcodebert(input):
   return output
 
 # Create and launch Gradio interface
-iface = gr.Interface(fn=use_graphcodebert, inputs=input, outputs=output)
+# Create and launch Gradio interface
+iface = gr.Interface.from_pretrained( # Use from_pretrained instead of from_pipeline
+    fn=use_graphcodebert, 
+    inputs=input, 
+    outputs=output,
+    model=model, # Pass the model as a parameter
+    tokenizer=tokenizer, # Pass the tokenizer as a parameter
+    title="GraphCodeBERT Code Synthesis", # Add a title for the web app
+    description="Enter a natural language query and get a code snippet generated by GraphCodeBERT.", # Add a description for the web app
+    examples=[["create a function that returns the sum of two numbers"], ["sort a list of numbers in ascending order"]], # Add some examples for the input
+    theme="huggingface" # Choose a theme for the web app
+)
 iface.launch()