Spaces:

ltg
/

no-en-translation

Sleeping

davda54 commited on Oct 5, 2023

Commit

cc76a88

1 Parent(s): f8494a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,6 @@
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
 from threading import Thread
 import gradio as gr

 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
+from transformers.generation import LogitsProcessor
 from threading import Thread
 import gradio as gr