ldhldh commited on
Commit
7e4d374
1 Parent(s): ca3a5e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -2,11 +2,7 @@ from threading import Thread
2
 
3
  import torch
4
  import gradio as gr
5
- from ctransformers import AutoModelForCausalLM
6
- from transformers import pipeline,AutoTokenizer, BertTokenizer, BertForSequenceClassification, StoppingCriteria, StoppingCriteriaList
7
- from peft import PeftModel, PeftConfig
8
  import re
9
- from kobert_transformers import get_tokenizer
10
 
11
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
12
  print("Running on device:", torch_device)
 
2
 
3
  import torch
4
  import gradio as gr
 
 
 
5
  import re
 
6
 
7
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
8
  print("Running on device:", torch_device)