debisoft commited on
Commit
2a0f9e2
1 Parent(s): 1b67131
Files changed (1) hide show
  1. app.py +28 -28
app.py CHANGED
@@ -41,67 +41,67 @@ def greet(input):
41
  "answer": "Collective bargaining of salary."
42
  })
43
 
44
- qa_list.append({
45
  "question": "How much are union dues, and what do they cover?",
46
  "answer": "The union dues for our union is 3%."
47
  })
48
 
49
- qa_list.append({
50
  "question": "How does the union handle grievances and disputes?",
51
  "answer": "There will be a panel to oversee disputes"
52
  })
53
 
54
- qa_list.append({
55
  "question": "Will joining a union affect my job security?",
56
  "answer": "No."
57
  })
58
 
59
- qa_list.append({
60
  "question": "What is the process for joining a union?",
61
  "answer": "Please use the contact form."
62
  })
63
 
64
- qa_list.append({
65
  "question": "How do unions negotiate contracts with employers?",
66
  "answer": "Our dear leader will handle the negotiations."
67
  })
68
 
69
- qa_list.append({
70
  "question": "What role do I play as a union member?",
71
  "answer": "You will be invited to our monthly picnics"
72
  })
73
 
74
- qa_list.append({
75
  "question": "How do unions ensure that employers comply with agreements?",
76
  "answer": "We will have a monthly meeting for members"
77
  })
78
 
79
- qa_list.append({
80
  "question": "Can I be forced to join a union?",
81
  "answer": "What kind of questions is that! Of course no!"
82
  })
83
 
84
- qa_list.append({
85
  "question": "What happens if I disagree with the union’s decisions?",
86
  "answer": "We will agree to disagree"
87
  })
88
 
89
- for qna in qa:
90
- ques_str = qna["question"]
91
- ans_str = qna["answer"]
92
- n_prompt_list.append(f"""
93
  <|im_start|>system\n{sys_str}<|im_end|>
94
  <|im_start|>question\n{ques_str}<|im_end|>
95
  <|im_start|>answer\n{ans}<|im_end|>
96
  """
97
- )
98
 
99
- n_prompt_str = "\n"
100
 
101
- for prompt in n_prompt_list:
102
- n_prompt_str = n_prompt_str + prompt + "\n"
103
 
104
- total_prompt=f"""
105
  {n_prompt_str}
106
 
107
  <|im_start|>system\n{sys_str}<|im_end|>
@@ -111,20 +111,20 @@ def greet(input):
111
  """
112
 
113
 
114
- print("***total_prompt:")
115
- print(total_prompt)
116
- response = get_completion(total_prompt)
117
- #gen_text = response["predictions"][0]["generated_text"]
118
- #return json.dumps(extract_json(gen_text, 3))
119
 
120
- ###gen_text = response["choices"][0]["text"]
121
 
122
- #return gen_text
123
 
124
- ###return json.dumps(extract_json(gen_text, -1))
125
- return response
126
 
127
- #return json.dumps(response)
128
 
129
  #iface = gr.Interface(fn=greet, inputs="text", outputs="text")
130
  #iface.launch()
 
41
  "answer": "Collective bargaining of salary."
42
  })
43
 
44
+ qa_list.append({
45
  "question": "How much are union dues, and what do they cover?",
46
  "answer": "The union dues for our union is 3%."
47
  })
48
 
49
+ qa_list.append({
50
  "question": "How does the union handle grievances and disputes?",
51
  "answer": "There will be a panel to oversee disputes"
52
  })
53
 
54
+ qa_list.append({
55
  "question": "Will joining a union affect my job security?",
56
  "answer": "No."
57
  })
58
 
59
+ qa_list.append({
60
  "question": "What is the process for joining a union?",
61
  "answer": "Please use the contact form."
62
  })
63
 
64
+ qa_list.append({
65
  "question": "How do unions negotiate contracts with employers?",
66
  "answer": "Our dear leader will handle the negotiations."
67
  })
68
 
69
+ qa_list.append({
70
  "question": "What role do I play as a union member?",
71
  "answer": "You will be invited to our monthly picnics"
72
  })
73
 
74
+ qa_list.append({
75
  "question": "How do unions ensure that employers comply with agreements?",
76
  "answer": "We will have a monthly meeting for members"
77
  })
78
 
79
+ qa_list.append({
80
  "question": "Can I be forced to join a union?",
81
  "answer": "What kind of questions is that! Of course no!"
82
  })
83
 
84
+ qa_list.append({
85
  "question": "What happens if I disagree with the union’s decisions?",
86
  "answer": "We will agree to disagree"
87
  })
88
 
89
+ for qna in qa:
90
+ ques_str = qna["question"]
91
+ ans_str = qna["answer"]
92
+ n_prompt_list.append(f"""
93
  <|im_start|>system\n{sys_str}<|im_end|>
94
  <|im_start|>question\n{ques_str}<|im_end|>
95
  <|im_start|>answer\n{ans}<|im_end|>
96
  """
97
+ )
98
 
99
+ n_prompt_str = "\n"
100
 
101
+ for prompt in n_prompt_list:
102
+ n_prompt_str = n_prompt_str + prompt + "\n"
103
 
104
+ total_prompt=f"""
105
  {n_prompt_str}
106
 
107
  <|im_start|>system\n{sys_str}<|im_end|>
 
111
  """
112
 
113
 
114
+ print("***total_prompt:")
115
+ print(total_prompt)
116
+ response = get_completion(total_prompt)
117
+ #gen_text = response["predictions"][0]["generated_text"]
118
+ #return json.dumps(extract_json(gen_text, 3))
119
 
120
+ ###gen_text = response["choices"][0]["text"]
121
 
122
+ #return gen_text
123
 
124
+ ###return json.dumps(extract_json(gen_text, -1))
125
+ return response
126
 
127
+ #return json.dumps(response)
128
 
129
  #iface = gr.Interface(fn=greet, inputs="text", outputs="text")
130
  #iface.launch()