Spaces:

KZMTx
/

SumbotZephyr3b

Sleeping

BenBranyon commited on Apr 12

Commit

578cce2

•

1 Parent(s): 65f8951

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,8 +19,8 @@ def query(payload):
     try:
         data = json.loads(response.content.decode("utf-8"))
         generated_text = data[0]['generated_text']
-        re.sub(r'<|system|>.*?<|assistant|>', '', generated_text, flags=re.DOTALL)
-        return response
     except:
         return response

     try:
         data = json.loads(response.content.decode("utf-8"))
         generated_text = data[0]['generated_text']
+        cleaned_text = re.sub(r'<|system|>.*?<|assistant|>', '', generated_text, flags=re.DOTALL)
+        return cleaned_text
     except:
         return response