Merge branch 'main' of https://huggingface.co/spaces/piyushgrover/PhiGPT into main

app.py

@@ -31,8 +31,8 @@ gen = pipeline('text-generation', model=model, tokenizer=tokenizer, max_length=3
 
 
 def fn_query_on_load():
-    prompt = "Explain nuclear physics to a five year old kid."
-    return prompt
+    input = "Explain nuclear physics to a five year old kid."
+    return input
 
 
 def generate_response(input):
@@ -40,8 +40,10 @@ def generate_response(input):
     prompt = f"### Human: {input}\n\n### Assistant: "
     result = gen(prompt)
 
-    resp = result[0]['generated_text'].replace(prompt, '')
-    resp_arr = resp.split('###')
+    resp = result[0]['generated_text']
+    print(resp)
+    resp_arr = resp.replace(prompt, '').split('###')
+    print(resp_arr)
 
     final_resp = resp_arr[0]