Duplicate from adxyz/cquip

Co-authored-by: Anthony Daniell <adxyz@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Cquip
+emoji: 📊
+colorFrom: red
+colorTo: yellow
+sdk: gradio
+sdk_version: 3.36.0
+app_file: app.py
+pinned: false
+duplicated_from: adxyz/cquip
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,65 @@
+import gradio as gr
+from transformers import pipeline
+
+model = pipeline("text-generation", model="PygmalionAI/pygmalion-1.3b", max_new_tokens=80)
+# model = pipeline("conversational", model="PygmalionAI/pygmalion-6b")
+# model = pipeline("conversational", model="PygmalionAI/pygmalion-1.3b", max_new_tokens=40)
+# model = pipeline("text-generation", model="PygmalionAI/pygmalion-2.7b")
+
+title = "Talk to Sheldon Cooper"
+description = """
+   <img src="https://tse1.mm.bing.net/th?id=OIP.dvsVHZcwIf9k1tA4WUmc5gHaEK&pid=Api&rs=1&c=1&qlt=95&w=169&h=95" width=200px> 
+   You can say things to Sheldon and get a Sheldonian quip in response.
+   """
+examples = [["Do you like comic books?"],["What is your favorite expression?"]]
+
+
+def predict(prompt):
+    full_prompt = ""
+    full_prompt += "Sheldon's persona: Sheldon is a narcissistic theoretical physicist. "   
+    full_prompt += "He lives in an apartment in the city of Pasadena in the state of California. " 
+#    full_prompt += "He loves comic books and hates sports. "
+#    full_prompt += "His best friends are Penny and Leonard. "
+#    full_prompt += "His favorite food is Chinese. "
+    full_prompt += "His favorite expression is Bazinga!"
+#    full_prompt += "\n<START>\n"
+    full_prompt += "\n"
+    full_prompt += "You: this looks like some serious work. Leonard did you do this?\n"
+    full_prompt += "Sheldon: Actually, that is my work.\n"
+    full_prompt = full_prompt + "You: " + prompt + "\n"
+    completion = model(full_prompt)[0]["generated_text"]
+
+    print('prompt: ')
+    print(prompt)
+    print('full_prompt: ')
+    print(full_prompt)
+    print('full response: ')
+    print(completion)
+
+# Filter for the output from Sheldon right after latest prompt from user.
+    parse_sentence = completion.split(sep="\n")
+    output_sentence="Sheldon: Bazinga-d!\n"  # Create a default in case of issues in the output.
+    
+    for iSent in range(len(parse_sentence)):
+        if len(parse_sentence[iSent]) == 0:  # Avoid junk.
+            break
+        print('parsed sentence number [ iSent ]: ', iSent)
+        print(parse_sentence[iSent])
+        print('len of parsed sentence = ', len(parse_sentence[iSent]))
+        print('parse_sentence[iSent][0] = ', parse_sentence[iSent][0])
+        print('parse_sentence[iSent][-1] = ', parse_sentence[iSent][-1])
+        print('prompt[0] = ', prompt[0])
+        print('prompt[-1] = ', prompt[-1])
+        if parse_sentence[iSent]==("You: " + prompt):
+            print('inside parse_sentence-prompt comparison')
+        # check for overruns and correct prefix
+            if iSent+1 < len(parse_sentence) and parse_sentence[iSent+1][0:8]=="Sheldon:":  
+                output_sentence=parse_sentence[iSent+1]
+            break
+    
+    return output_sentence
+
+iface = gr.Interface(fn=predict, inputs="text", outputs="text", 
+                     title=title, description=description,
+                    examples=examples)
+iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+torch
+transformers