Spaces:
Sleeping
Sleeping
Commit ·
6a8c601
1
Parent(s): b09b3fc
Update app.py
Browse files
app.py
CHANGED
|
@@ -84,7 +84,7 @@ model.summary()
|
|
| 84 |
model.compile(loss = 'categorical_crossentropy', optimizer = 'adam', metrics = ['accuracy'])
|
| 85 |
|
| 86 |
#executing the model
|
| 87 |
-
history = model.
|
| 88 |
|
| 89 |
# plotting the loss
|
| 90 |
plt.plot(history.history['loss'],label = 'train_loss')
|
|
@@ -154,4 +154,4 @@ def greet_user(name):
|
|
| 154 |
return image_output_class
|
| 155 |
|
| 156 |
app = gd.Interface(fn = greet_user, inputs='image', outputs='text')
|
| 157 |
-
app.launch(
|
|
|
|
| 84 |
model.compile(loss = 'categorical_crossentropy', optimizer = 'adam', metrics = ['accuracy'])
|
| 85 |
|
| 86 |
#executing the model
|
| 87 |
+
history = model.fit(train_set, validation_data = (validate_set), epochs = 5, verbose = 1)
|
| 88 |
|
| 89 |
# plotting the loss
|
| 90 |
plt.plot(history.history['loss'],label = 'train_loss')
|
|
|
|
| 154 |
return image_output_class
|
| 155 |
|
| 156 |
app = gd.Interface(fn = greet_user, inputs='image', outputs='text')
|
| 157 |
+
app.launch(debug=True)
|