dar-tau commited on
Commit
0b8d742
·
verified ·
1 Parent(s): 74d1efc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,8 +1,9 @@
1
  import os
 
 
2
  import gradio as gr
3
  import spaces
4
  import torch
5
- from typing import Optional
6
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
7
  from dataclasses import dataclass
8
 
 
1
  import os
2
+ from typing import Optional
3
+ from functools import partial
4
  import gradio as gr
5
  import spaces
6
  import torch
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
8
  from dataclasses import dataclass
9