Spaces:
Sleeping
Sleeping
Commit ·
c435edf
1
Parent(s): a2316a0
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,17 +64,17 @@ model_q_path = "facebook/opt-125m_3bit"
|
|
| 64 |
|
| 65 |
|
| 66 |
generator = pipeline('text-generation', model="facebook/opt-125m_3bit")
|
| 67 |
-
generator_q = pipeline('text-generation', model="facebook/opt-125m-awq")
|
| 68 |
|
| 69 |
|
| 70 |
def generate_text_pip(prompt):
|
| 71 |
generated_text = generator(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
|
| 72 |
return generated_text
|
| 73 |
-
|
| 74 |
def generate_text_pip_q(prompt):
|
| 75 |
generated_text = generator_q(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
|
| 76 |
return generated_text
|
| 77 |
-
|
| 78 |
print(generator("I went to boston and"))
|
| 79 |
print("quantized model",generator_q("I went to boston and"))
|
| 80 |
|
|
@@ -100,10 +100,10 @@ def generate_text_from_quantized(prompt):
|
|
| 100 |
# Create a Gradio interface
|
| 101 |
iface = gr.Interface(fn=generate_text_pip, inputs="text", outputs="text")
|
| 102 |
|
| 103 |
-
iface_2 = gr.Interface(fn=generate_text_pip_q, inputs="text", outputs="text")
|
| 104 |
-
|
| 105 |
|
| 106 |
-
|
|
|
|
| 107 |
|
| 108 |
# Launch the Gradio app
|
| 109 |
-
app.launch()
|
|
|
|
| 64 |
|
| 65 |
|
| 66 |
generator = pipeline('text-generation', model="facebook/opt-125m_3bit")
|
| 67 |
+
#generator_q = pipeline('text-generation', model="facebook/opt-125m-awq")
|
| 68 |
|
| 69 |
|
| 70 |
def generate_text_pip(prompt):
|
| 71 |
generated_text = generator(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
|
| 72 |
return generated_text
|
| 73 |
+
'''
|
| 74 |
def generate_text_pip_q(prompt):
|
| 75 |
generated_text = generator_q(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
|
| 76 |
return generated_text
|
| 77 |
+
'''
|
| 78 |
print(generator("I went to boston and"))
|
| 79 |
print("quantized model",generator_q("I went to boston and"))
|
| 80 |
|
|
|
|
| 100 |
# Create a Gradio interface
|
| 101 |
iface = gr.Interface(fn=generate_text_pip, inputs="text", outputs="text")
|
| 102 |
|
| 103 |
+
#iface_2 = gr.Interface(fn=generate_text_pip_q, inputs="text", outputs="text")
|
|
|
|
| 104 |
|
| 105 |
+
iface.launch()
|
| 106 |
+
#app = gr.TabbedInterface([iface, iface_2],["Normal", "Quantized"])
|
| 107 |
|
| 108 |
# Launch the Gradio app
|
| 109 |
+
#app.launch()
|