daniel-dona commited on
Commit
2705c5c
·
verified ·
1 Parent(s): fd78eab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import spaces
2
  import gradio as gr
3
- from transformers import pipeline
4
 
5
  model_name = "daniel-dona/gemma-3-270m-it"
6
 
 
1
  import spaces
2
  import gradio as gr
3
+ from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
  model_name = "daniel-dona/gemma-3-270m-it"
6