atwine commited on
Commit
318def7
·
1 Parent(s): 1dab9af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import time
3
- from ctransformers import AutoModelForCausalLM # Please ensure this import is correct
4
 
5
  PROMPT_TEMPLATE = (
6
  "<s>" "[INST]"
 
1
  import gradio as gr
2
  import time
3
+ from transformers import AutoModelForCausalLM # Please ensure this import is correct
4
 
5
  PROMPT_TEMPLATE = (
6
  "<s>" "[INST]"