parth parekh commited on
Commit
fa5ad45
·
1 Parent(s): 3a154fd

added same schema for both

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -50,7 +50,8 @@ async def detect_contact(input: TextInput):
50
  is_contact = predict(input.text) # You can adjust this threshold as needed
51
  return {
52
  "text": input.text,
53
- "is_contact_info": is_contact,
 
54
  "method": "model"
55
  }
56
  except Exception as e:
 
50
  is_contact = predict(input.text) # You can adjust this threshold as needed
51
  return {
52
  "text": input.text,
53
+ "contact_probability": 0.98,
54
+ "is_contact_info": is_contact == 1,
55
  "method": "model"
56
  }
57
  except Exception as e: