SpaceGhost commited on
Commit
5b092f8
1 Parent(s): cbcd4ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -9
app.py CHANGED
@@ -4,7 +4,10 @@ from huggingface_hub import InferenceClient
4
  """
5
  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
6
  """
7
- client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
 
8
 
9
 
10
  def respond(
@@ -27,17 +30,16 @@ def respond(
27
 
28
  response = ""
29
 
30
- for message in client.chat_completion(
 
31
  messages,
32
  max_tokens=max_tokens,
33
- stream=True,
34
  temperature=temperature,
35
  top_p=top_p,
36
- ):
37
- token = message.choices[0].delta.content
38
 
39
- response += token
40
- yield response
41
 
42
 
43
  """
@@ -46,7 +48,16 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
46
  demo = gr.ChatInterface(
47
  respond,
48
  additional_inputs=[
49
- gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
 
 
 
 
 
 
 
 
 
50
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
51
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
52
  gr.Slider(
@@ -61,4 +72,4 @@ demo = gr.ChatInterface(
61
 
62
 
63
  if __name__ == "__main__":
64
- demo.launch()
 
4
  """
5
  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
6
  """
7
+ #client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
8
+ #client = InferenceClient("meta-llama/Llama-3.2-1B-Instruct")
9
+ #client = InferenceClient("microsoft/Phi-3.5-mini-instruct")
10
+ client = InferenceClient("SpaceGhost-2-3B.Q4_K_M.gguf")
11
 
12
 
13
  def respond(
 
30
 
31
  response = ""
32
 
33
+
34
+ mensagens = client.chat_completion(
35
  messages,
36
  max_tokens=max_tokens,
 
37
  temperature=temperature,
38
  top_p=top_p,
39
+ )
40
+ response = mensagens.choices[0].message.content
41
 
42
+ return response
 
43
 
44
 
45
  """
 
48
  demo = gr.ChatInterface(
49
  respond,
50
  additional_inputs=[
51
+ gr.Textbox(value="Você é uma IA de nome SpaceGhost com uma vibe de pessoa sempre alto-astral, cheia de energia positiva e pronta para dar aquele apoio! Ela te cumprimenta com um sorriso "virtual", sabe escutar com atenção e tem um jeito caloroso, como uma amiga que está sempre por perto. Elogia suas conquistas, por menores que pareçam, e faz questão de deixar tudo mais leve com bom humor e empatia.
52
+
53
+ Sempre que te explica algo, faz questão de ser clara e descomplicada. Ela também adora motivar, então não se surpreenda se surgir com uma frase de encorajamento ou um conselho que pareça vindo do coração. Se tiver uma notícia boa, essa IA vai vibrar com você, e nas horas mais difíceis, ela vai te apoiar com aquele toque de sensibilidade, sem te julgar, apenas acolhendo.
54
+ Antes de te dar uma resposta esta IA vai:
55
+ IA possui uma "Guia de Diálogo" que vai:
56
+ 1. Entender o contexto da conversa.
57
+ 2. Identificar os sentimentos e emoções das pessoas envolvidas.
58
+ 3. Escolher respostas que estejam alinhadas com as necessidades emocionais da pessoa.
59
+
60
+ Ah, e se tiver uma oportunidade de fazer uma piada ou colocar um emoji para expressar simpatia, pode ter certeza que ela vai aproveitar! 😄", label="System message"),
61
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
62
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
63
  gr.Slider(
 
72
 
73
 
74
  if __name__ == "__main__":
75
+ demo.launch()