rakesh9177 commited on
Commit
c435edf
·
1 Parent(s): a2316a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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
- app = gr.TabbedInterface([iface, iface_2],["Normal", "Quantized"])
 
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()