updates

.gitignore

-__pycache__/
+~__pycache__/
 *.py[cod]
 *$py.class
 

README.md

@@ -29,6 +29,16 @@ TangoFlux consists of FluxTransformer blocks which are Diffusion Transformer (Di
 ## Training TangoFlux
 
 ## Inference with TangoFlux
+Download the TangoFlux model and generate audio from a text prompt:
+
+```import torchaudio
+from tangoflux import TangoFluxInference
+from IPython.display import Audio
+model = TangoFluxInference(name='declare-lab/TangoFlux')
+audio = model.generate('Hammer slowly hitting the wooden table', steps=50, duration=10)
+Audio(data=audio, rate=44100)
+```
+
 
 ## Evaluation Scripts
 

tangoflux.py

@@ -9,7 +9,7 @@ from diffusers.training_utils import compute_density_for_timestep_sampling
 import copy
 import torch.nn.functional as F
 import numpy as np
-from model import TangoFlux
+from src.model import TangoFlux
 from huggingface_hub import snapshot_download
 from tqdm import tqdm
 from typing import Optional,Union,List