rphrp1985 commited on
Commit
c3adc49
1 Parent(s): a4804b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -1
app.py CHANGED
@@ -136,7 +136,32 @@ async def get_answer(q: QueryM ):
136
  except Exception as e:
137
  print(e)
138
  return JSONResponse({"status":False})
139
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
140
 
141
 
142
 
 
136
  except Exception as e:
137
  print(e)
138
  return JSONResponse({"status":False})
139
+
140
+
141
+ @app.post("/openchat35")
142
+ async def get_answer(q: QueryM ):
143
+ text = q.text
144
+ try:
145
+ client = InferenceClient()
146
+ generate_kwargs = dict(
147
+ max_new_tokens= int(q.tokens),
148
+ repetition_penalty=1.0,
149
+ top_p= q.topp,
150
+ top_k=int(q.topk),
151
+ temperature=q.temp,
152
+ stop= ["<|end_of_turn|>", "<??|>"]
153
+ )
154
+ inputs= text
155
+ response = client.post(json={"inputs": inputs, "parameters": generate_kwargs},model="openchat/openchat_3.5")
156
+ json_string = response.decode('utf-8')
157
+ list_of_dicts = json.loads(json_string)
158
+ result_dict = list_of_dicts[0]
159
+ x=(result_dict['generated_text'])
160
+ x=x.replace(inputs,'')
161
+ return JSONResponse({"result":x,"status":True})
162
+ except Exception as e:
163
+ print(e)
164
+ return JSONResponse({"status":False})
165
 
166
 
167