Keshav-rejoice commited on
Commit
ee59f36
·
verified ·
1 Parent(s): 0835bcd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -6
app.py CHANGED
@@ -10,25 +10,35 @@ import os
10
 
11
  openai.api_key = os.getenv('OPENAI_API_KEY')
12
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  class_labels = [
14
  "Algae",
15
- "Bubbles and blisters",
16
  "Cracks",
17
- "Efflorescence",
18
  "Fungus",
19
- "Patchiness",
20
  "Peeling",
21
- "Poor Hiding",
22
- "Shade Variation"
23
  ]
24
 
 
25
  def encode_image(image_path):
26
  with open(image_path, "rb") as image_file:
27
  return base64.b64encode(image_file.read()).decode("utf-8")
28
 
29
  @st.cache_resource
30
  def load_trained_model():
31
- return load_model('my_model12.h5')
 
32
 
33
  loaded_model = load_trained_model()
34
 
 
10
 
11
  openai.api_key = os.getenv('OPENAI_API_KEY')
12
 
13
+ # class_labels = [
14
+ # "Algae",
15
+ # "Bubbles and blisters",
16
+ # "Cracks",
17
+ # "Efflorescence",
18
+ # "Fungus",
19
+ # "Patchiness",
20
+ # "Peeling",
21
+ # "Poor Hiding",
22
+ # "Shade Variation"
23
+ # ]
24
+
25
  class_labels = [
26
  "Algae",
 
27
  "Cracks",
28
+ "Effloroscence"
29
  "Fungus",
 
30
  "Peeling",
 
 
31
  ]
32
 
33
+
34
  def encode_image(image_path):
35
  with open(image_path, "rb") as image_file:
36
  return base64.b64encode(image_file.read()).decode("utf-8")
37
 
38
  @st.cache_resource
39
  def load_trained_model():
40
+ # return load_model('my_model12.h5')
41
+ return load_model('my_new_model12.h5')
42
 
43
  loaded_model = load_trained_model()
44