Spaces:

joheras
/

suicide-intent

Runtime error

joheras commited on Oct 17, 2023

Commit

68a18d6

1 Parent(s): 6710de1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,5 @@
-from transformers import pipeline
 import gradio as gr
 model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1",torch_dtype=torch.float16, device_map="auto")

+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 import gradio as gr
 model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1",torch_dtype=torch.float16, device_map="auto")