prafullcodes commited on
Commit
6a8c601
·
1 Parent(s): b09b3fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.fit_generator(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,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(share=True,debug=True)
 
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)