paavansundar commited on
Commit
0ed65fe
1 Parent(s): 3b45b65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -77,7 +77,9 @@ def queryGPT(question):
77
 
78
  def generate_response(model,tokenizer, prompt, max_length=200):
79
  input_ids = tokenizer.encode(prompt, return_tensors="pt") # 'pt' for returning pytorch tensor
80
-
 
 
81
  # Create the attention mask and pad token id
82
  attention_mask = torch.ones_like(input_ids)
83
  pad_token_id = tokenizer.eos_token_id
@@ -99,6 +101,6 @@ with gr.Blocks() as demo:
99
  btn = gr.Button(value="Submit")
100
  btn.click(queryGPT, inputs=[txt_input], outputs=[txt_output])
101
  if __name__ == "__main__":
102
- prepareData()
103
- fine_tune_gpt()
104
  demo.launch()
 
77
 
78
  def generate_response(model,tokenizer, prompt, max_length=200):
79
  input_ids = tokenizer.encode(prompt, return_tensors="pt") # 'pt' for returning pytorch tensor
80
+ #my_model = GPT2LMHeadModel.from_pretrained(model_output_path)
81
+ #my_tokenizer = GPT2Tokenizer.from_pretrained(model_output_path)
82
+
83
  # Create the attention mask and pad token id
84
  attention_mask = torch.ones_like(input_ids)
85
  pad_token_id = tokenizer.eos_token_id
 
101
  btn = gr.Button(value="Submit")
102
  btn.click(queryGPT, inputs=[txt_input], outputs=[txt_output])
103
  if __name__ == "__main__":
104
+ #prepareData()
105
+ #fine_tune_gpt()
106
  demo.launch()