boluobobo commited on
Commit
78f9cea
·
verified ·
1 Parent(s): 8c5d898

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -76,6 +76,7 @@ def predict(image: Image.Image):
76
  api_output = {
77
  "ai_probability": round(ai_prob, 3),
78
  "human_probability": round(human_prob, 3),
 
79
  "top3_sources": top3_sources
80
  }
81
 
 
76
  api_output = {
77
  "ai_probability": round(ai_prob, 3),
78
  "human_probability": round(human_prob, 3),
79
+ "predicted_source": predicted_source,
80
  "top3_sources": top3_sources
81
  }
82