Update app.py

app.py

@@ -1,14 +1,13 @@
-
-pip install transformers
-pip install huggingface_hub
+# Install required libraries
+pip install transformers huggingface_hub gradio torch
+# Import necessary modules
 from huggingface_hub import login
-login(" ")
-
-pip install gradio transformers torch
-
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Load the tokenizer and model
+# Log in to Hugging Face (replace 'your_token' with your actual Hugging Face token)
+login("your_huggingface_token")
+
+# Load the tokenizer and model from Hugging Face
 tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen-350M-multi")
 model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-350M-multi")
 
@@ -29,4 +28,5 @@ generated_ids = model.generate(
 # Decode the generated tokens to text
 generated_code = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
 
-print(generated_code)
+# Output the generated code
+print(generated_code)