Update app.py
Browse files
app.py
CHANGED
@@ -113,71 +113,13 @@ For more information on `huggingface_hub` Inference API support, please check th
|
|
113 |
# if __name__ == "__main__":
|
114 |
# demo.launch()
|
115 |
|
116 |
-
# import gradio as gr
|
117 |
-
# from huggingface_hub import InferenceClient
|
118 |
-
|
119 |
-
# """
|
120 |
-
# For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
|
121 |
-
# """
|
122 |
-
# client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")
|
123 |
-
|
124 |
-
# def respond(message, history: list[tuple[str, str]]):
|
125 |
-
# system_message = (
|
126 |
-
# "You are a helpful and experienced coding assistant specialized in web development. "
|
127 |
-
# "Help the user by generating complete and functional code for building websites. "
|
128 |
-
# "You can provide HTML, CSS, JavaScript, and backend code (like Flask, Node.js, etc.) based on their requirements. "
|
129 |
-
# "Break down the tasks clearly if needed, and be friendly and supportive in your responses."
|
130 |
-
# )
|
131 |
-
# max_tokens = 2048
|
132 |
-
# temperature = 0.7
|
133 |
-
# top_p = 0.95
|
134 |
-
|
135 |
-
# messages = [{"role": "system", "content": system_message}]
|
136 |
-
|
137 |
-
# for val in history:
|
138 |
-
# if val[0]:
|
139 |
-
# messages.append({"role": "user", "content": val[0]})
|
140 |
-
# if val[1]:
|
141 |
-
# messages.append({"role": "assistant", "content": val[1]})
|
142 |
-
|
143 |
-
# messages.append({"role": "user", "content": message})
|
144 |
-
|
145 |
-
# response = ""
|
146 |
-
|
147 |
-
# for message in client.chat_completion(
|
148 |
-
# messages,
|
149 |
-
# max_tokens=max_tokens,
|
150 |
-
# stream=True,
|
151 |
-
# temperature=temperature,
|
152 |
-
# top_p=top_p,
|
153 |
-
# ):
|
154 |
-
# token = message.choices[0].delta.content
|
155 |
-
|
156 |
-
# response += token
|
157 |
-
# yield response
|
158 |
-
|
159 |
-
# """
|
160 |
-
# For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
|
161 |
-
# """
|
162 |
-
# demo = gr.ChatInterface(respond)
|
163 |
-
|
164 |
-
# if __name__ == "__main__":
|
165 |
-
# demo.launch()
|
166 |
-
|
167 |
import gradio as gr
|
168 |
from huggingface_hub import InferenceClient
|
169 |
-
import os
|
170 |
|
171 |
"""
|
172 |
-
For more information on `huggingface_hub` Inference API support, please check the docs:
|
173 |
-
https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
|
174 |
"""
|
175 |
-
|
176 |
-
# Load token from environment variable named 'sdp'
|
177 |
-
token = os.getenv("sdp")
|
178 |
-
|
179 |
-
# Initialize the Inference Client with the secret token
|
180 |
-
client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct", token=token)
|
181 |
|
182 |
def respond(message, history: list[tuple[str, str]]):
|
183 |
system_message = (
|
@@ -209,13 +151,13 @@ def respond(message, history: list[tuple[str, str]]):
|
|
209 |
temperature=temperature,
|
210 |
top_p=top_p,
|
211 |
):
|
212 |
-
|
213 |
-
|
|
|
214 |
yield response
|
215 |
|
216 |
"""
|
217 |
-
For information on how to customize the ChatInterface, peruse the gradio docs:
|
218 |
-
https://www.gradio.app/docs/chatinterface
|
219 |
"""
|
220 |
demo = gr.ChatInterface(respond)
|
221 |
|
@@ -224,3 +166,4 @@ if __name__ == "__main__":
|
|
224 |
|
225 |
|
226 |
|
|
|
|
113 |
# if __name__ == "__main__":
|
114 |
# demo.launch()
|
115 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
116 |
import gradio as gr
|
117 |
from huggingface_hub import InferenceClient
|
|
|
118 |
|
119 |
"""
|
120 |
+
For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
|
|
|
121 |
"""
|
122 |
+
client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")
|
|
|
|
|
|
|
|
|
|
|
123 |
|
124 |
def respond(message, history: list[tuple[str, str]]):
|
125 |
system_message = (
|
|
|
151 |
temperature=temperature,
|
152 |
top_p=top_p,
|
153 |
):
|
154 |
+
token = message.choices[0].delta.content
|
155 |
+
|
156 |
+
response += token
|
157 |
yield response
|
158 |
|
159 |
"""
|
160 |
+
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
|
|
|
161 |
"""
|
162 |
demo = gr.ChatInterface(respond)
|
163 |
|
|
|
166 |
|
167 |
|
168 |
|
169 |
+
|