R-PRM-Demo

Running on Zero

kevinpro commited on 20 days ago

Commit

734f1a7

verified ·

1 Parent(s): 31a4eef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import spaces
 import gradio as gr
 from sacremoses import MosesPunctNormalizer
 from stopes.pipelines.monolingual.utils.sentence_split import get_split_algo
-from transformers import AutoTokenizer, AutoModel
 from flores import code_mapping
 import platform
 import torch

 import gradio as gr
 from sacremoses import MosesPunctNormalizer
 from stopes.pipelines.monolingual.utils.sentence_split import get_split_algo
+from transformers import AutoTokenizer, AutoModel,AutoModelForCausalLM
 from flores import code_mapping
 import platform
 import torch