Update app.py

app.py

@@ -60,16 +60,41 @@
 # chatbot = gr.ChatInterface(YourCoder_chatbot, title="YourCoder Chatbot", description="Enter piece of code to generate a code review!")
 # chatbot.launch()
 
+# import gradio as gr
+
+# # def YourCoder_chatbot(message, history):
+# #     gr.load("models/bigcode/starcoder")
+
+# # chatbot = gr.ChatInterface(YourCoder_chatbot, title="YourCoder Chatbot", description="Enter piece of code to generate a code review!")
+# chatbot = gr.Interface(fn=gr.load("models/bigcode/starcoder"), inputs=[gr.Textbox(label="Insert Code Snippet",lines=5)],
+#                     outputs=[gr.Textbox(label="Review Here",lines=8)],
+#                     title="Code Reviewer"
+#                     )
+# # gr.load("models/bigcode/starcoder").launch()
+
+# chatbot.launch()
+
+
+#####################
+import os
 import gradio as gr
+from transformers import pipeline
 
-# def YourCoder_chatbot(message, history):
-#     gr.load("models/bigcode/starcoder")
+# Get the token from environment variables
+# token = os.getenv("HUGGINGFACE_TOKEN")
+# if token is None:
+#     raise ValueError("Hugging Face token is not set in the environment variables.")
 
-# chatbot = gr.ChatInterface(YourCoder_chatbot, title="YourCoder Chatbot", description="Enter piece of code to generate a code review!")
-chatbot = gr.Interface(fn=gr.load("models/bigcode/starcoder"), inputs=[gr.Textbox(label="Insert Code Snippet",lines=5)],
-                    outputs=[gr.Textbox(label="Review Here",lines=8)],
-                    title="Code Reviewer"
-                    )
-# gr.load("models/bigcode/starcoder").launch()
+# Load the model from the Hugging Face Model Hub with authentication
+generator = pipeline('text-generation', model='bigcode/starcoder', use_auth_token=token)
+
+# Define the prediction function
+def generate_text(prompt):
+    result = generator(prompt, max_length=50)
+    return result[0]['generated_text']
+
+# Create the Gradio interface
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
 
-chatbot.launch()
+# Launch the app
+iface.launch()