Spaces:
Sleeping
Sleeping
paavansundar
commited on
Commit
•
0ed65fe
1
Parent(s):
3b45b65
Update app.py
Browse files
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()
|