Spaces:

Deepak107
/

Img_Net

Sleeping

Deepak107 commited on Aug 28, 2022

Commit

08a0f9a

1 Parent(s): 44200e0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,14 +1,14 @@
-import tensorflow
 import gradio
 import requests
-inception_net = tensorflow.keras.applications.MobileNetV2()
 response = requests.get("https://git.io/JJkYN")
 labels = response.text.split("\n")
 def classify_image(inp):
   inp = inp.reshape((-1, 224, 224, 3))
-  inp = tensorflow.keras.applications.mobilenet_v2.preprocess_input(inp)
   prediction = inception_net.predict(inp).flatten()
   confidences = {labels[i]: float(prediction[i]) for i in range(1000)}
   return confidences

+from tensorflow import keras
 import gradio
 import requests
+inception_net = keras.applications.MobileNetV2()
 response = requests.get("https://git.io/JJkYN")
 labels = response.text.split("\n")
 def classify_image(inp):
   inp = inp.reshape((-1, 224, 224, 3))
+  inp = keras.applications.mobilenet_v2.preprocess_input(inp)
   prediction = inception_net.predict(inp).flatten()
   confidences = {labels[i]: float(prediction[i]) for i in range(1000)}
   return confidences