Spaces:

Yashnik
/

code-review

Runtime error

Yashnik commited on Jun 25, 2024

Commit

eb89b8b

verified ·

1 Parent(s): 961ad1b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,10 +1,11 @@
 from transformers import pipeline, Conversation
 import gradio as gr
 import os
-os.environ['HUGGINGFACE_API_KEY'] = os.getenv("HUGGINGFACE_API_KEY")
-chatbot = pipeline(model="bigcode/starcoder")
 message_list = []
 response_list = []

 from transformers import pipeline, Conversation
 import gradio as gr
 import os
+from getpass import getpass
+model = os.getenv('bigcode/starcoder')
+chatbot = pipeline(model=model)
 message_list = []
 response_list = []