ghosthets commited on
Commit
c11c991
·
verified ·
1 Parent(s): d8f9409

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
- client = InferenceClient("ghothets/dex") # 👈 Replace with your actual model ID
5
 
6
  def respond(message, history, system_message, max_tokens, temperature, top_p):
7
  prompt = ""
@@ -9,8 +9,8 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
9
  prompt += f"User: {user}\nDex: {bot}\n"
10
  prompt += f"User: {message}\nDex:"
11
 
 
12
  try:
13
- response = ""
14
  for token in client.text_generation(
15
  prompt,
16
  max_new_tokens=max_tokens,
@@ -26,8 +26,8 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
26
  demo = gr.ChatInterface(
27
  fn=respond,
28
  additional_inputs=[
29
- gr.Textbox(value="You are Dex AI 🤖", label="System Prompt"),
30
- gr.Slider(1, 2048, value=512, step=1, label="Max New Tokens"),
31
  gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Temperature"),
32
  gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p"),
33
  ],
@@ -36,4 +36,4 @@ demo = gr.ChatInterface(
36
  )
37
 
38
  if __name__ == "__main__":
39
- demo.launch(share=True)
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
+ client = InferenceClient("dexcommunity/dex") # 👈 Apna Dex model ka path
5
 
6
  def respond(message, history, system_message, max_tokens, temperature, top_p):
7
  prompt = ""
 
9
  prompt += f"User: {user}\nDex: {bot}\n"
10
  prompt += f"User: {message}\nDex:"
11
 
12
+ response = ""
13
  try:
 
14
  for token in client.text_generation(
15
  prompt,
16
  max_new_tokens=max_tokens,
 
26
  demo = gr.ChatInterface(
27
  fn=respond,
28
  additional_inputs=[
29
+ gr.Textbox(value="You are Dex AI 🤖", label="System Message"),
30
+ gr.Slider(1, 2048, value=512, step=1, label="Max Tokens"),
31
  gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Temperature"),
32
  gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p"),
33
  ],
 
36
  )
37
 
38
  if __name__ == "__main__":
39
+ demo.launch()