dar-tau commited on
Commit
3dbd18e
1 Parent(s): 25d6129

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,12 +4,13 @@ from typing import Optional
4
  from dataclasses import dataclass
5
  from copy import deepcopy
6
  from functools import partial
 
7
  import spaces
8
  import gradio as gr
9
  import torch
10
  from datasets import load_dataset
11
- from ctransformers import AutoModelForCausalLM as CAutoModelForCausalLM
12
  from transformers import PreTrainedModel, PreTrainedTokenizer, AutoModelForCausalLM, AutoTokenizer
 
13
  from interpret import InterpretationPrompt
14
  from configs import model_info, dataset_info
15
 
 
4
  from dataclasses import dataclass
5
  from copy import deepcopy
6
  from functools import partial
7
+ import numpy as np
8
  import spaces
9
  import gradio as gr
10
  import torch
11
  from datasets import load_dataset
 
12
  from transformers import PreTrainedModel, PreTrainedTokenizer, AutoModelForCausalLM, AutoTokenizer
13
+ from ctransformers import AutoModelForCausalLM as CAutoModelForCausalLM
14
  from interpret import InterpretationPrompt
15
  from configs import model_info, dataset_info
16