anirudh-sub commited on
Commit
95d1712
·
1 Parent(s): 33bfed7

Debugged app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -22,6 +22,13 @@ import torch
22
  import gradio as gr
23
 
24
  model = "anirudh-sub/debate_model_v2.1"
 
 
 
 
 
 
 
25
 
26
  def debate_response(text):
27
  sequences = pipeline(
@@ -32,7 +39,6 @@ def debate_response(text):
32
  eos_token_id=tokenizer.eos_token_id,
33
  max_length=500,
34
  )
35
-
36
  response = ""
37
  for seq in sequences:
38
  reponse += {seq['generated_text']}
 
22
  import gradio as gr
23
 
24
  model = "anirudh-sub/debate_model_v2.1"
25
+ tokenizer = AutoTokenizer.from_pretrained(model)
26
+ pipeline = transformers.pipeline(
27
+ "text-generation",
28
+ model=model,
29
+ torch_dtype=torch.float16,
30
+ device_map="auto",
31
+ )
32
 
33
  def debate_response(text):
34
  sequences = pipeline(
 
39
  eos_token_id=tokenizer.eos_token_id,
40
  max_length=500,
41
  )
 
42
  response = ""
43
  for seq in sequences:
44
  reponse += {seq['generated_text']}