alfredplpl commited on
Commit
b19740e
·
verified ·
1 Parent(s): 4f5738b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,8 +3,9 @@
3
  import gradio as gr
4
  import os
5
  import spaces
6
- from transformers import GemmaTokenizer, AutoModelForCausalLM
7
- from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
8
  from threading import Thread
9
 
10
 
 
3
  import gradio as gr
4
  import os
5
  import spaces
6
+ import torch
7
+ from transformers import AutoTokenizer, AutoModelForCausalLM
8
+ from transformers import TextIteratorStreamer
9
  from threading import Thread
10
 
11