Spaces:

timothytampold
/

StyleGAN3-playground

Runtime error

timothytampold commited on Nov 6, 2023

Commit

b1087a6

1 Parent(s): 2bd0717

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
 import gradio as gr
-from transformers import AutoFeatureExtractor, AutoModelForImageGeneration
 feature_extractor = AutoFeatureExtractor.from_pretrained("akhaliq/Stylegan-ffhq-vintage")
-model = AutoModelForImageGeneration.from_pretrained("akhaliq/Stylegan-ffhq-vintage")
 def generate_image(seed):
     inputs = feature_extractor(seed, return_tensors="pt")

 import gradio as gr
+from transformers import AutoFeatureExtractor, TFAutoModel
 feature_extractor = AutoFeatureExtractor.from_pretrained("akhaliq/Stylegan-ffhq-vintage")
+model = TFAutoModel.from_pretrained("akhaliq/Stylegan-ffhq-vintage")
 def generate_image(seed):
     inputs = feature_extractor(seed, return_tensors="pt")