Spaces:

mareloraby
/

meter2poem-1

Build error

mareloraby commited on Sep 7, 2022

Commit

2ac6809

1 Parent(s): 8ca0f96

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ bertSharedM = EncoderDecoderModel.from_pretrained("mareloraby/BERTShared-meter2p
 def generate_response(text, k = 70, p = 0.9, nb = 4):
 # meters = set(['الرمل','البسيط','الخفيف','الكامل','السريع','الطويل','المتقارب','الرجز','المجتث','المنسرح','الوافر','المقتضب','الهزج','المديد','المضارع'])
   prompt = f"{text}"
-  encoded_prompt = tokenizerM.encode_plus(prompt, return_tensors = 'pt').to(device)
   gneration = bertSharedM.generate(
                             input_ids = encoded_prompt.input_ids,
                             attention_mask = encoded_prompt.attention_mask,

 def generate_response(text, k = 70, p = 0.9, nb = 4):
 # meters = set(['الرمل','البسيط','الخفيف','الكامل','السريع','الطويل','المتقارب','الرجز','المجتث','المنسرح','الوافر','المقتضب','الهزج','المديد','المضارع'])
   prompt = f"{text}"
+  encoded_prompt = tokenizerM.encode_plus(prompt, return_tensors = 'pt')#.to(device)
   gneration = bertSharedM.generate(
                             input_ids = encoded_prompt.input_ids,
                             attention_mask = encoded_prompt.attention_mask,