Spaces:

GenAIQE
/

APIGenAI

Running

vishwas3086 commited on Feb 13

Commit

160808e

verified ·

1 Parent(s): b936450

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import sqlite3
 from transformers import pipeline
 # Load AI model for test case generation
-test_case_generator = pipeline("text-generation", model="microsoft/CodeGPT-small")
 def fetch_swagger_data(swagger_url):
     try:

 from transformers import pipeline
 # Load AI model for test case generation
+test_case_generator = pipeline("text-generation", model="microsoft/CodeGPT-small-py")
 def fetch_swagger_data(swagger_url):
     try: