Update app.py

app.py

@@ -1,21 +1,14 @@
 import gradio as gr
+import os
 from transformers import pipeline
 
+access_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
-# Initialize the StarCoder model
-# Replace 'models/bigcode/starcoder' with the actual model path if different
-code_generator = pipeline('text-generation', model='google/codegemma-2b')
 
-def generate_code(prompt):
-    #"""
-    #Generates code based on the given prompt.
+code_generator = pipeline('text-generation', model='google/codegemma-2b', token=access_token)
 
-    #Parameters:
-    #- prompt (str): The input text to generate code from.
+def generate_code(prompt):
 
-    #Returns:
-    #- str: The generated code.
-    #"""
     generated_code = code_generator(prompt, max_length=100, do_sample=True, temperature=0.7)
     return generated_code[0]['generated_text']
 
@@ -24,8 +17,8 @@ iface = gr.Interface(
     fn=generate_code,
     inputs=gr.Textbox(lines=5, label="Input Text"),
     outputs=gr.Textbox(label="Generated Code"),
-    title="StarCoder Code Generator",
-    description="Enter a prompt to generate code using the StarCoder model.",
+    title="Gemma Code Generator",
+    description="Enter a prompt to generate code using the little gemma model.",
 )
 
 # Launch the Gradio interface