danhergir commited on
Commit
b934874
1 Parent(s): aa0a746

fixes added

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. utils.py +1 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import streamlit as st
 
2
 
3
  from utils import load_model, generate
4
 
 
1
  import streamlit as st
2
+ import torch
3
 
4
  from utils import load_model, generate
5
 
utils.py CHANGED
@@ -1,6 +1,6 @@
1
  import numpy as np
2
  import torch
3
- from huggan.pytorch.lightweight_gan.lighweight_gan import LightWeightGAN
4
 
5
  def load_model(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
6
  gan = LightWeightGAN.from_pretrained(model_name, version=model_version)
 
1
  import numpy as np
2
  import torch
3
+ from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
4
 
5
  def load_model(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
6
  gan = LightWeightGAN.from_pretrained(model_name, version=model_version)