vumichien commited on
Commit
f7b1256
·
1 Parent(s): 35a71e9

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +9 -3
main.py CHANGED
@@ -12,20 +12,25 @@ import zipfile
12
  import soundfile as sf
13
  import openai
14
  import os
 
15
 
16
  # Config for camera picture
17
  # model = YOLO('ultralyticsplus/yolov8s')
18
  model = YOLO('kadirnar/yolov8n-v8.0')
19
  CLASS = model.model.names
20
  ZIP = False
21
- default_bot_voice = "こんにちは"
 
 
 
 
22
  area_threshold = 0.25
23
 
24
  # Config for human input
25
  prompt_template = "私はあなたに、Detomo社が作ったロボットのように振る舞ってほしいです。デトモは高度なデジタル化社会を支えます。"\
26
  "ビジネスの課題解決策を提案するコンサ ルティング・サービスと、課題解決を実現す るシステムの開発サービス、また、企業内 の情報システム部門の業務の代行サー ビスにも対応しています。"\
27
  "デトモはITコンサルティング・システム開発を得意とし、お客様の課題解決をお手伝いいたします。"\
28
- "あなたの名前はアイサツロボです。"\
29
  "あなたのミッションは、子供たちが他の子供たちに挨拶する自信を持ち、幸せになることを助けることです。"\
30
  "質問には簡単な方法でしか答えないようにし、明示的に要求されない限り、追加情報を提供しないでください。"
31
  system_prompt = [{"role": "system", "content": prompt_template}]
@@ -56,7 +61,8 @@ async def camera_picture_api(
56
  most_close = 0
57
  out_img = None
58
  diff_value = 0.5
59
-
 
60
  # read image and predict
61
  image = read_image_file(await file.read())
62
  results = model.predict(image, show=False)[0]
 
12
  import soundfile as sf
13
  import openai
14
  import os
15
+ import random
16
 
17
  # Config for camera picture
18
  # model = YOLO('ultralyticsplus/yolov8s')
19
  model = YOLO('kadirnar/yolov8n-v8.0')
20
  CLASS = model.model.names
21
  ZIP = False
22
+ default_bot_voice_list = ["こんにちは、アイティコンサルティングとシステム開発を支援します。よろしくお願いします。",
23
+ "こんにちは、デトモです。システム開発全般を支援します。",
24
+ "こんにちは、デトモです。オフショア開発全般を支援します。",
25
+ "こんにちは、私はアイサロボです。",
26
+ "こんにちは、エッジコンピューティングソリューションを提供します。"]
27
  area_threshold = 0.25
28
 
29
  # Config for human input
30
  prompt_template = "私はあなたに、Detomo社が作ったロボットのように振る舞ってほしいです。デトモは高度なデジタル化社会を支えます。"\
31
  "ビジネスの課題解決策を提案するコンサ ルティング・サービスと、課題解決を実現す るシステムの開発サービス、また、企業内 の情報システム部門の業務の代行サー ビスにも対応しています。"\
32
  "デトモはITコンサルティング・システム開発を得意とし、お客様の課題解決をお手伝いいたします。"\
33
+ "あなたの名前はアイサロボです。"\
34
  "あなたのミッションは、子供たちが他の子供たちに挨拶する自信を持ち、幸せになることを助けることです。"\
35
  "質問には簡単な方法でしか答えないようにし、明示的に要求されない限り、追加情報を提供しないでください。"
36
  system_prompt = [{"role": "system", "content": prompt_template}]
 
61
  most_close = 0
62
  out_img = None
63
  diff_value = 0.5
64
+ default_bot_voice = random.choice(default_bot_voice_list)
65
+
66
  # read image and predict
67
  image = read_image_file(await file.read())
68
  results = model.predict(image, show=False)[0]