Hugging Face's logo

Spaces:
Xalphinions
/
watermelon
Sleeping

App Files Community
Xalphinions commited on
Commit
89cf057
·
verified ·
1 Parent(s): 50b800e

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch, torchaudio
2
  import argparse
3
  from preprocess import process_audio_data, process_image_data
4
- from train_2 import WatermelonModel
5
 
6
 
7
  def infer(audio, image, model, device):
 
1
  import torch, torchaudio
2
  import argparse
3
  from preprocess import process_audio_data, process_image_data
4
+ from train import WatermelonModel
5
 
6
 
7
  def infer(audio, image, model, device):