Spaces:

openai
/

openai-detector

Running on CPU Upgrade

App Files Files Community

openai-detector / post_endpoint.patch

XciD HF staff

Dockerize + pip freeze

7c4081b over 1 year ago

raw history blame contribute delete

No virus

4.02 kB

	��diff --git a/detector/server.py b/detector/server.py
	index 34a0c85..8c190e4 100644
	--- a/detector/server.py
	+++ b/detector/server.py
	@@ -20,6 +20,22 @@ def log(*args):

	class RequestHandler(SimpleHTTPRequestHandler):

	+ def do_POST(self):
	+ content_length = int(self.headers['Content-Length']) # <--- Gets the size of data
	+ post_data = self.rfile.read(content_length) # <--- Gets the data itself
	+ post_data = json.loads(post_data.decode('utf-8'))
	+
	+ self.begin_content('application/json;charset=UTF-8')
	+
	+ all_tokens, used_tokens, real, fake = self.infer(post_data['text'])
	+
	+ self.wfile.write(json.dumps(dict(
	+ all_tokens=all_tokens,
	+ used_tokens=used_tokens,
	+ real_probability=real,
	+ fake_probability=fake
	+ )).encode())
	+
	def do_GET(self):
	query = unquote(urlparse(self.path).query)

	@@ -32,6 +48,16 @@ class RequestHandler(SimpleHTTPRequestHandler):

	self.begin_content('application/json;charset=UTF-8')

	+ all_tokens, used_tokens, real, fake = self.infer(query)
	+
	+ self.wfile.write(json.dumps(dict(
	+ all_tokens=all_tokens,
	+ used_tokens=used_tokens,
	+ real_probability=real,
	+ fake_probability=fake
	+ )).encode())
	+
	+ def infer(query):
	tokens = tokenizer.encode(query)
	all_tokens = len(tokens)
	tokens = tokens[:tokenizer.max_len - 2]
	@@ -45,12 +71,7 @@ class RequestHandler(SimpleHTTPRequestHandler):

	fake, real = probs.detach().cpu().flatten().numpy().tolist()

	- self.wfile.write(json.dumps(dict(
	- all_tokens=all_tokens,
	- used_tokens=used_tokens,
	- real_probability=real,
	- fake_probability=fake
	- )).encode())
	+ return all_tokens, used_tokens, real, fake

	def begin_content(self, content_type):
	self.send_response(200)