Spaces:

diamantrsd
/

copywriting-otomatis

Running

diamantrsd commited on May 10

Commit

5547ccf

•

1 Parent(s): ed89481

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,8 +16,8 @@ except Exception as e:
     print(f"An error occurred: {e}")
 # Load GPT-2 tokenizer and model
-tokenizer = GPT2Tokenizer.from_pretrained("diamantrsd/copywriting-otomatis-gpt2")
-gpt2_model = GPT2LMHeadModel.from_pretrained("diamantrsd/copywriting-otomatis-gpt2")
 def predict_image(img_path, keyword=""):
     if img_path is None:

     print(f"An error occurred: {e}")
 # Load GPT-2 tokenizer and model
+tokenizer = GPT2Tokenizer.from_pretrained("diamantrsd/copywriting-otomatis")
+gpt2_model = GPT2LMHeadModel.from_pretrained("diamantrsd/copywriting-otomatis")
 def predict_image(img_path, keyword=""):
     if img_path is None: