Александр Романов commited on
Commit
8737f8d
·
1 Parent(s): a6eb639

Text correction

Browse files

Remove plasma from options

Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -17,7 +17,9 @@ model.add(Dense(output_len, activation='sigmoid', bias_initializer='zeros'))
17
  model_checkpoint_path = os.path.join('model', 'biofilms_model_checkpoint')
18
  model.load_weights(model_checkpoint_path)
19
 
20
- materials_labels = ["плазма", "фтор", "графен", "Si/SiO2: 2 - нано Si", "Al/Al(OH)3", "Li", "Ti/TiO2", "Pb", "ZrO2", "Co", "V", "Fe/FeOn: 2 - ферралит", "Cu", "Ag", "Cr", "Zn", "ZnS", "Mn", "Ca", "Mg", "Na", "K", "NH4/NO2", "CO2 atm", "солонец", "Muscovite", "Mont-t", "Kaolinite", "Chlorite", "Fillite", "Clinochlor", "Calcite", "Dolomite", "Barite", "Gypsum"]
 
 
21
  sharpness_values = ["волнистая", "пористая", "дисперсная", "грубодисперсная"]
22
 
23
  def get_sample_dat(sample):
@@ -45,7 +47,7 @@ def calculate_result(materials, sharpness, angles):
45
  predict = model.predict(np.expand_dims(X, axis=0), verbose=0)[0]
46
  result = (predict[0] / (predict[0] + predict[1])) * 100.0
47
  result_string = "{:.1f}".format(result)
48
- return f"Biofilm exist chance: {result_string} / {predict[0]} / {predict[1]} / {X}%"
49
 
50
  def greet(Materials, Sharpness, Angles):
51
  return "Hello, my friend " + Angles + "!! /=/ " + str(Materials[0]) + " /=/ " + str(Sharpness)
@@ -55,6 +57,7 @@ def button_click(Materials, Sharpness, Angles):
55
  sharpness = 0
56
  angles = []
57
  try:
 
58
  for var in range(len(materials_labels)):
59
  if (var in Materials):
60
  materials.append(1.0)
@@ -73,7 +76,7 @@ def button_click(Materials, Sharpness, Angles):
73
  predict = model.predict(np.expand_dims(X, axis=0),verbose = 0)[0]
74
  result = (predict[0]/(predict[0]+predict[1]))*100.0
75
  result_string = "{:.1f}".format(result)
76
- return f"Biofilm exist chance: {result_string}%"
77
  except ValueError as e:
78
  return f"Error: {e}"
79
 
@@ -81,8 +84,8 @@ iface = gr.Interface(
81
  fn=button_click,
82
  inputs = [
83
  gr.CheckboxGroup(choices=materials_labels, label="Materials", type='index'),
84
- gr.Radio(sharpness_values, label="Sharpness", type='index'),
85
- gr.Textbox(lines=1, label="Write contanc angles:")
86
  ],
87
  outputs="text")
88
 
 
17
  model_checkpoint_path = os.path.join('model', 'biofilms_model_checkpoint')
18
  model.load_weights(model_checkpoint_path)
19
 
20
+ materials_labels = ["фтор", "графен", "Si/SiO2", "Al/Al(OH)3", "Li", "Ti/TiO2", "Pb", "ZrO2", "Co", "V", "Fe/FeOn",
21
+ "Cu", "Ag", "Cr", "Zn", "ZnS", "Mn", "Ca", "Mg", "Na", "K", "NH4/NO2", "CO2 atm", "солонец", "Muscovite",
22
+ "Mont-t", "Kaolinite", "Chlorite", "Fillite", "Clinochlor", "Calcite", "Dolomite", "Barite", "Gypsum"]
23
  sharpness_values = ["волнистая", "пористая", "дисперсная", "грубодисперсная"]
24
 
25
  def get_sample_dat(sample):
 
47
  predict = model.predict(np.expand_dims(X, axis=0), verbose=0)[0]
48
  result = (predict[0] / (predict[0] + predict[1])) * 100.0
49
  result_string = "{:.1f}".format(result)
50
+ return f"Biofilm existence chance: {result_string} / {predict[0]} / {predict[1]} / {X}%"
51
 
52
  def greet(Materials, Sharpness, Angles):
53
  return "Hello, my friend " + Angles + "!! /=/ " + str(Materials[0]) + " /=/ " + str(Sharpness)
 
57
  sharpness = 0
58
  angles = []
59
  try:
60
+ materials.append(0.0) # Plasma value
61
  for var in range(len(materials_labels)):
62
  if (var in Materials):
63
  materials.append(1.0)
 
76
  predict = model.predict(np.expand_dims(X, axis=0),verbose = 0)[0]
77
  result = (predict[0]/(predict[0]+predict[1]))*100.0
78
  result_string = "{:.1f}".format(result)
79
+ return f"Biofilm existence chance: {result_string}%"
80
  except ValueError as e:
81
  return f"Error: {e}"
82
 
 
84
  fn=button_click,
85
  inputs = [
86
  gr.CheckboxGroup(choices=materials_labels, label="Materials", type='index'),
87
+ gr.Radio(sharpness_values, label="Roughness", type='index'),
88
+ gr.Textbox(lines=1, label="Write contact angles:")
89
  ],
90
  outputs="text")
91