saltacc commited on
Commit
dbede30
1 Parent(s): 7730934

use remote repo

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
6
 
7
- model = AutoModelForCausalLM.from_pretrained("salt/RandomPrompt-v1")
8
 
9
  model.to(device)
10
 
@@ -26,5 +26,5 @@ def detect(text_in, max_length):
26
  return text
27
 
28
 
29
- iface = gr.Interface(fn=detect, inputs=[gr.Textbox(), gr.Slider(50, 200, default=120)], outputs=gr.TextArea())
30
  iface.launch()
 
4
 
5
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
6
 
7
+ model = AutoModelForCausalLM.from_pretrained("saltacc/RandomPrompt-v1")
8
 
9
  model.to(device)
10
 
 
26
  return text
27
 
28
 
29
+ iface = gr.Interface(fn=detect, inputs=[gr.Textbox(), gr.Slider(100, 200, default=120)], outputs=gr.TextArea())
30
  iface.launch()