anirudh-sub commited on
Commit
b45ed0e
·
1 Parent(s): 417c80a

Trying to debug app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -2,6 +2,14 @@ import gradio as gr
2
  import transformers
3
  from transformers import AutoTokenizer
4
 
 
 
 
 
 
 
 
 
5
  def greet(name):
6
  return "Hello " + name + "!"
7
 
 
2
  import transformers
3
  from transformers import AutoTokenizer
4
 
5
+ model = "anirudh-sub/debate_model_v2.1"
6
+ tokenizer = AutoTokenizer.from_pretrained(model)
7
+ pipeline = transformers.pipeline(
8
+ "text-generation",
9
+ model=model,
10
+ device_map="auto",
11
+ )
12
+
13
  def greet(name):
14
  return "Hello " + name + "!"
15