spycoder commited on
Commit
83f90bc
1 Parent(s): 1ba2548

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -46,11 +46,11 @@ The model was trained on Thai audio recordings with the following sentences so p
46
  def predict(file_upload,microphone):
47
  max_length = 100000
48
  file_path =file_upload
49
- warn_output = " "
50
  if (microphone is not None) and (file_upload is not None):
51
  warn_output = (
52
  "WARNING: You've uploaded an audio file and used the microphone. "
53
- "The recorded file from the microphone will be used and the uploaded audio will be discarded.\n"
54
  )
55
 
56
  elif (microphone is None) and (file_upload is None):
@@ -76,7 +76,7 @@ def predict(file_upload,microphone):
76
  logits = logits.squeeze()
77
  predicted_class_id = torch.argmax(logits, dim=-1).item()
78
 
79
- return warn_output + "\n" + "You probably have SP" if predicted_class_id == 1 else warn_output + "\n" + "You probably don't have SP"
80
  gr.Interface(
81
  fn=predict,
82
  inputs=[
 
46
  def predict(file_upload,microphone):
47
  max_length = 100000
48
  file_path =file_upload
49
+ warn_output = ""
50
  if (microphone is not None) and (file_upload is not None):
51
  warn_output = (
52
  "WARNING: You've uploaded an audio file and used the microphone. "
53
+ "The recorded file from the microphone will be used and the uploaded audio will be discarded.\n\n"
54
  )
55
 
56
  elif (microphone is None) and (file_upload is None):
 
76
  logits = logits.squeeze()
77
  predicted_class_id = torch.argmax(logits, dim=-1).item()
78
 
79
+ return warn_output + "You probably have SP" if predicted_class_id == 1 else warn_output + "You probably don't have SP"
80
  gr.Interface(
81
  fn=predict,
82
  inputs=[