Tonic commited on
Commit
d7c2816
·
verified ·
1 Parent(s): bcbfe89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
- import gemma
4
- from gemma.modeling_gemma import GemmaForCausalLM
5
  import torch
6
  import time
7
 
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
+ from transformers.gemma.modeling_gemma import GemmaForCausalLM
 
4
  import torch
5
  import time
6