Update for clarifai to image identification

speks7 · Mar 2, 2018 · 45d07c7 · 45d07c7
1 parent 6ed1ca1
commit 45d07c7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/src/components/AnswerNotification/index.js b/src/components/AnswerNotification/index.js
@@ -4,7 +4,7 @@ import PropTypes from "prop-types";
 
 import styles from "./styles";
 
-const ThingToFind = "Pizza";
+//const ThingToFind = "Pizza";
 
 class AnswerNotification extends Component {
   constructor(props) {
@@ -23,7 +23,7 @@ class AnswerNotification extends Component {
   }
 
   render() {
-    const isValid = this.props.answer === ThingToFind;
+    const isValid = this.props.answer === this.props.answer;
     const icon = isValid
       ? require("../../assets/valid.png")
       : require("../../assets/error.png");

diff --git a/src/screens/PredictScreen/index.js b/src/screens/PredictScreen/index.js
@@ -29,7 +29,7 @@ class PredictScreen extends Component {
 
   componentDidMount() {
     const clarifai = new Clarifai.App({
-      apiKey: "b0959ce79c6e4b6eb7af2c91fb561046" //dummy
+      apiKey: "" //dummy
     });
 
     process.nextTick = setImmediate; // RN polyfill
@@ -41,26 +41,23 @@ class PredictScreen extends Component {
       .predict(Clarifai.GENERAL_MODEL, file)
       .then(response => {
         const { concepts } = response.outputs[0].data;
-
+        
         if (concepts && concepts.length > 0) {
           for (const prediction of concepts) {
-            if (prediction.name === "pizza" && prediction.value >= 0.99) {
-              return this.setState({ loading: false, result: "Pizza" });
-            }
-            this.setState({ result: "Not Pizza" });
+              return this.setState({ loading: false, result: prediction.name });
           }
         }
-
-        this.setState({ loading: false });
       })
+      .catch(err => alert(err));
+      /*
       .catch(e => {
         Alert.alert(
           "An error has occurred",
           "Sorry, the quota may be exceeded, try again later!",
           [{ text: "OK", onPress: () => this._cancel() }],
           { cancelable: false }
         );
-      });
+      });*/
   }
 
   _cancel() {