Spaces:

dominguezdaniel
/

image-post

Sleeping

dominguezdaniel commited on Mar 6, 2024

Commit

88ea942

verified ·

1 Parent(s): 41c9ebe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,6 @@ def generate_tweet(label):
     inputs = tokenizer.encode(prompt, return_tensors="pt")
     outputs = model.generate(inputs, max_length=280, num_return_sequences=1, no_repeat_ngram_size=2)
     tweet = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    tweet = tweet_raw.replace("Write a tweet about", "")
     return tweet
 def predict(image):

     inputs = tokenizer.encode(prompt, return_tensors="pt")
     outputs = model.generate(inputs, max_length=280, num_return_sequences=1, no_repeat_ngram_size=2)
     tweet = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return tweet
 def predict(image):