Spaces:

RWTHMaterials
/

DP800_DamageClassification

Sleeping

kerzel commited on Sep 12, 2025

Commit

80ba11b

1 Parent(s): 45a9fe7

load converted keras models

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ DEFAULT_IMAGE_PATH = "SE_001_cut.png"
 model1_windowsize = [250,250]
 #model1_threshold = 0.7
-model1 = keras.models.load_model('rwthmaterials_dp800_network1_inclusion.h5')
 model1.compile()
 damage_classes = {3: "Martensite",2: "Interface",0:"Notch",1:"Shadowing"}
@@ -31,7 +31,7 @@ damage_classes = {3: "Martensite",2: "Interface",0:"Notch",1:"Shadowing"}
 model2_windowsize = [100,100]
 #model2_threshold = 0.5
-model2 = keras.models.load_model('rwthmaterials_dp800_network2_damage.h5')
 model2.compile()

 model1_windowsize = [250,250]
 #model1_threshold = 0.7
+model1 = keras.models.load_model('rwthmaterials_dp800_network1_inclusion.keras')
 model1.compile()
 damage_classes = {3: "Martensite",2: "Interface",0:"Notch",1:"Shadowing"}
 model2_windowsize = [100,100]
 #model2_threshold = 0.5
+model2 = keras.models.load_model('rwthmaterials_dp800_network2_damage.keras')
 model2.compile()